2019 카카오 개발자 겨울 인턴십 크레인 인형뽑기 게임 level1

 

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

문제를 살펴보면 위에서부터 쌓고 위에서부터 없어지는 걸 확인하면 Queue 자료구조를 떠올렸다.

 

그리고 크레인으로 뽑은 인형을 넣는 바구니는 Stack 자료구조로 만들었다.

 

import java.util.*;

class Solution {
    public int solution(int[][] board, int[] moves) {
        int answer = 0;
        Map<Integer, Queue<Integer>> map = new HashMap<>();
        Stack<Integer> stack = new Stack<>();

        for(int i = 0;i<board.length;i++) {
            map.put(i+1,new LinkedList<>());
        }

        for(int i = 0;i<board.length;i++) {
            for(int j =0;j<board[i].length;j++){
                System.out.print(board[i][j] + " ");
                if(board[i][j] == 0) {
                    continue;
                }
                map.get(j+1).add(board[i][j]);
            }
            System.out.println();
        }

        for(int move : moves) {
            if(map.get(move).isEmpty()) {
                continue;
            }
            int doll = map.get(move).poll();
            if(!stack.isEmpty() && stack.peek() == doll) {
                answer += 2;
                stack.pop();
                continue;
            }

            stack.push(doll);

        }
        return answer;
    }
}