[Java] 프로그래머스 - 크레인 인형뽑기 게임

[문제]

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

프로그래머스

 

[풀이]

스택을 이용하여 풀이를 진행하였습니다.

만약 board의 크기가 N 이라면, N + 1 크기의 스택배열 Stack을 선언합니다.

1~N까지 인형뽑기 기계 모양을 stack[1] ~ stack[N] 에 저장하고, 뽑힌 인형들을 담는 임시 stack은 stack[0]을 이용합니다.

 

스택은 FIFO의 특성을 지니고 있기 때문에 for문을 통해 스택에 board배열의 밑줄부터 차례로 인형 number를 저장합니다.이때, 만약 0이라면 칸이 비어있으므로 continue해줍니다.

 

그다음, moves배열 크기만큼 pick()를 호출하여 인형을 한개씩 뽑아줍니다.

pick() 함수에 파라미터로 인형칸의 index인 moves의 숫자를 전달하여 해당줄에서 인형을 뽑아 임시 stack에 push합니다.

이때, push를 진행하기에 앞서 임시 stack의 마지막 값을 peek하여 만약 뽑힌 인형과 같다면 상쇄하고 answer값을 갱신하고 같지 않다면 그대로 push합니다.

 

[코드]

import java.util.*;
class Solution {
    private static Stack<Integer> stack[]; // 0은 뽑아서 저장하는곳, 1 ~ N은 뽑기 기계
    private static int answer = 0;
    public int solution(int[][] board, int[] moves) {
        int N = board.length;
        stack = new Stack[N + 1];
        for(int i=0;i<=N;i++) {
            stack[i] = new Stack<Integer>();
        }
        for(int i=N-1;i>=0;i--) {
            for(int j=0;j<N;j++) {
                if(board[i][j] == 0) {
                    continue;
                }
                stack[j+1].push(board[i][j]);
            }
        }
        for(int i=0;i<moves.length;i++) {
            pick(moves[i]);
        }
        return answer;
    }
    public static void pick(int index) {
        if(stack[index].isEmpty()) {
            return;
        }
        int p = stack[index].pop();
        if(stack[0].isEmpty()) {
            stack[0].push(p);
            return;
        }
        if(stack[0].peek() == p) {
            stack[0].pop();
            answer += 2;
            return;
        }
        stack[0].push(p);
    }
}