Programmers 크레인 인형뽑기 게임 C++

2022-04-04 131 words One minute

Table of Contents

접근 방법

Stack 자료구조를 활용해서 쉽게 해결할 수 있습니다. 입력으로 주어진 moves 배열의 컬럼 값이 1부터 시작한다는 점에 유의합니다.

코드

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#include <bits/stdc++.h>

using namespace std;

int solution(vector<vector<int>> board, vector<int> moves) {
    int answer = 0;
    stack<int> stk;
    for (auto col : moves) {
        int target = 0;
        for (int i = 0; i < board.size(); ++i) {
            if (board[i][col-1]) {
                target = board[i][col-1];
                board[i][col-1] = 0;
                break;
            }
        }
        
        if (target == 0) {
            continue;
        }
        
        if (!stk.empty() and stk.top() == target) {
            stk.pop();
            answer += 2;
            continue;
        }
        
        stk.push(target);
    }
    return answer;
}