[Programmers / Level 1] 64061. 크레인 인형뽑기 / 2019 카카오 개발자 겨울 인턴십 (Java)

💡 Info

• 난이도: Lv1
  프로그래머스 링크: https://school.programmers.co.kr/learn/courses/30/lessons/64061
  풀이 링크(GitHub): hayannn/CodingTest_Java/프로그래머스/1/크레인 인형뽑기

입출력 조건

입출력 예시

문제 이해

알고리즘

풀이 시간 : 49분

• 인형이 있으면 -> 해당 칸은 0으로 설정하기
• 스택 사용
  • 같은 인형이면 제거하고, 터트린 인형 개수를 증가
  • 인형이 바구니 맨 위와 다를 경우에도 -> 인형을 바구니에 넣기

import java.util.*;

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

        for (int move : moves) {
            move -= 1;

            for (int i = 0; i < board.length; i++) {
                int doll = board[i][move];

                if (doll != 0) {
                    board[i][move] = 0;

                    if (!basket.isEmpty() && basket.peek() == doll) {
                        basket.pop();
                        answer += 2;
                    } else {
                        basket.push(doll);
                    }
                    break;
                }
            }
        }

        return answer;
    }
}

결과