프로그래머스 lv1 크레인 인형 뽑기

문제 링크

크레인 인형 뽑기

풀이

import java.util.Stack;

class Solution {
    public int solution(int[][] board, int[] moves) {
        int answer = 0;

        Stack<Integer> stack = new Stack<>();
        for (int i : moves) {
            for (int[] ints : board) {
                if (ints[i - 1] != 0) {
                    if (!stack.isEmpty() && stack.peek() == ints[i - 1]) {
                        stack.pop();
                        answer++;
                    } else {
                        stack.add(ints[i - 1]);
                    }
                    ints[i-1] = 0;
                    break;
                }
            }
        }
        return answer * 2;
    }
}

소감

• 처음으로 한번에 풀었다.
• stack 쓰면 너무 간단하게 풀리는데 이거 맞나?
• 암튼 앞서 풀었던 문제들에서 얻은 지식이 도움이 되었다. 기분 좋네