[프로그래머스/JAVA 자바] 크레인 인형뽑기 게임

 

프로그래머스

문제풀이

1. 인형을 뽑아서 바구니에 넣는데, 바닥에서부터 차곡차곡 쌓임 ( Stack 사용 )

2. 같은 인형이 바구니에 겹치면 제거 ( Stack 의 pop 와 peek 활용 )

 

나의풀이 ( 코드 )

import java.util.*;

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

        for (int move : moves) {

            for (int j = 0; j < board.length; j++) {
                if (board[j][move - 1] != 0) {
                    if (!stack.isEmpty()) {
                        if (stack.peek() == board[j][move - 1]) {
                            board[j][move - 1] = 0;
                            answer += 2;
                            stack.pop();
                            break;
                        }
                    }
                    stack.push(board[j][move - 1]);
                    board[j][move - 1] = 0;
                    break;
                }
            }
        }
        return answer;
    }
}