크레인 인형뽑기 게임

class Solution {
    public int solution(int[][] board, int[] moves) {
        int answer = 0;
        
        int[] tops = new int[board.length], stack = new int[moves.length];;
        int i, j, top = -1, pos = 0;
        boolean flag = true;
 
        // top 정의
        for (i = 0; i < board.length; i++) {
            for (j = 0; j < board.length; j++)
                if (board[j][i] != 0) {
                    tops[i] = j;
                    break;
                }
        }
        // stack에 쌓기
        for (i = 0; i < moves.length; i++){
            if (tops[moves[i] - 1] > board.length-1)
                continue;
            else
                stack[++top] = board[tops[moves[i] - 1]++][moves[i] - 1];
       }
        while (flag) {
            flag = false;
            for (i = 0; i < stack.length - 1; i++) {
                if(stack[i+1] == 0) {
                    flag = false;
                    break;
                }
                    
                if(stack[i] != -1)
                    pos = i;
            
                if (stack[pos] == stack[i + 1] && stack[pos] != -1) {
                    answer += 2;
                    stack[pos] = stack[i + 1] = -1;
                    flag=true;
                    break;
                }
            }
 
        }
        return answer;
    }
}

자바 자료구조 중에 기본적으로 stack과 list를 제공해준다.

이를 이용하면 더 간단하게 풀 수 있다.

 

각 board 마다의 꼭대기를 설정해주고, 이를 이용하여 stack을 쌓아준다.

-1을 건너뛰면서 같은 수가 연속으로 나올 경우 answer를 더해준다.