크레인 인형뽑기 게임

 

programmers.co.kr/learn/courses/30/lessons/64061

코딩테스트 연습 - 크레인 인형뽑기 게임

 

import java.util.*;
class Solution {
    public int solution(int[][] boards, int[] moves) {
        int cnt = 0;   
        Stack<Integer> stack = new Stack<>();
        /*
        * 1. 각자리에 인덱스로 값을 꺼내온다.
          2. 스택에 넣는다
          3. 스택에 peek해서 같은 값이면 cnt +한다.
        */
        for (int move : moves) {
            int index = move -1;
            
            // 전체 배열에서 하나씩 꺼낸다.
            for (int[] board : boards) {
                int doll = board[index];
                if (doll > 0) {
                    
                    if (!stack.isEmpty() && stack.peek() == doll){
                        stack.pop();
                        cnt += 2;
                    } else {
                        stack.push(doll);    
                    }
                    board[index] = 0;
                    // 다음배열은 검사하지않게 break
                    break;
                }
            }
        }
         
        return cnt;
    }
}

moves를 살펴보면 index에 +1 한값이라는 것을 알수있다. 

board를 하나씩 꺼내와서 인덱스에 해당하는 값과 stack의 값을 비교한다.

있으면 pop으로 제거, 없으면 스택에 넣는다.