[프로그래머스] 크레인 인형뽑기 게임

프로그래머스 level1 크레인 인형뽑기 게임

 

소요시간 : 8분

 

그냥 문제가 하라고 하는대로 짜면 되는 문제였다

문제에서 말하는 바구니를 list로 하고 list.get 으로 했는데

stack 으로 peek 해서 짜는 게 훨씬 편할 듯!

 

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int solution(int[][] board, int[] moves) {
    	int N = board.length;
    	int answer = 0;
    	List<Integer> list = new ArrayList<>();
    	
    	for (int i = 0; i < moves.length; i++) {
    		for (int i2 = 0; i2 < N; i2++) {
				if(board[i2][moves[i]-1] == 0) continue;
				
				list.add(board[i2][moves[i]-1]);
				board[i2][moves[i]-1] = 0;
				break;
			}
    		if(list.size() > 1 && list.get(list.size()-1) == list.get(list.size()-2)) {
    			list.remove(list.size()-1);
    			list.remove(list.size()-1);
    			answer += 2;
    		}
		}
    	
        return answer;
    }
}