크레인 인형뽑기 게임

개굴이·2023년 9월 21일
0

코딩테스트

목록 보기
26/58
post-thumbnail

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

게임 화면의 격자의 상태가 담긴 2차원 배열 board와 인형을 집기 위해 크레인을 작동시킨 위치가 담긴 배열 moves가 매개변수로 주어질 때, 크레인을 모두 작동시킨 후 터트려져 사라진 인형의 개수를 return 하도록 solution 함수를 완성해주세요.

제한사항

board 배열은 2차원 배열로 크기는 "5 x 5" 이상 "30 x 30" 이하입니다.
board의 각 칸에는 0 이상 100 이하인 정수가 담겨있습니다.
0은 빈 칸을 나타냅니다.
1 ~ 100의 각 숫자는 각기 다른 인형의 모양을 의미하며 같은 숫자는 같은 모양의 인형을 나타냅니다.
moves 배열의 크기는 1 이상 1,000 이하입니다.
moves 배열 각 원소들의 값은 1 이상이며 board 배열의 가로 크기 이하인 자연수입니다.

소스

import java.util.*;

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

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

        return answer;
    }
}

1개의 댓글

comment-user-thumbnail
2023년 9월 21일

덕분에 Stack의 존재를 알았어용 🐳

답글 달기