[알고리즘] 5-3 크레인 인형뽑기(카카오) - Stack (인프런 자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 강의)

인프런의 자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 강의를 수강하며 간략한 문제 설명, 예습 풀이 코드, 강의에서 설명하는 풀이 코드를 정리하고 있습니다.

• 문제 링크 : https://cote.inflearn.com/contest/10/problem/05-03
• 문제 설명 : https://programmers.co.kr/learn/courses/30/lessons/64061

예습 풀이

• Stack 자료구조를 사용하여 풀이

package inflearn.stack_queue;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;

//크레인 인형뽑기(카카오)
public class Main5_3 {

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

        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                int pos = board[j][moves[i] - 1]; //크레인 위치
                board[j][moves[i] - 1] = 0;

                if (pos != 0) {
                    if (!stack.isEmpty() && stack.peek() == pos) {
                        stack.pop();
                        answer += 2;
                    } else {
                        stack.push(pos);
                    }
                    break;
                }
            }
        }

        return answer;
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        int[][] board = new int[n][n];
        StringTokenizer st = null;

        for (int i = 0; i < n; i++) {
             st = new StringTokenizer(br.readLine());

            for (int j = 0; st.hasMoreTokens(); j++) {
                board[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        int m = Integer.parseInt(br.readLine());
        int[] moves = new int[m];

        st = new StringTokenizer(br.readLine());
        for (int i = 0; st.hasMoreTokens(); i++) {
            moves[i] = Integer.parseInt(st.nextToken());
        }

        System.out.println(solution(board, n, moves, m));
    }
}

복습 풀이

• 예습 풀이와 강의에서 설명하는 풀이 방식이 동일하기 때문에 복습 풀이 생략