[JAVA/알고리즘] 크레인 인형뽑기 게임

 

https://programmers.co.kr/learn/courses/30/lessons/64061?language=java 

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

 

 

import java.util.*;

class Solution {
    public int solution(int[][] board, int[] moves) {
        int answer = 0;
        ArrayList<Integer> basket = new ArrayList<Integer>();

        for (int m : moves) {
            // m-1번 위치에서 인형 집기
            for (int i = 0; i < board.length; ++i) {
                if (board[i][m-1] != 0) {
                    // 바구니에 옮겨 담기
                    if (basket.isEmpty()) { // 바구니가 비어있는 경우
                        basket.add(board[i][m-1]);
                    } else if (board[i][m-1] == basket.get(basket.size()-1)) {  // 인형이 터지는 경우
                        basket.remove(basket.size()-1);
                        answer += 2;
                    } else {    // 바구니에 담기는 경우
                        basket.add(board[i][m-1]);
                    }
                    board[i][m-1] = 0;
                    break;
                }
            }
        }

        return answer;
    }
}