[C++/PGS] Lv.1 : 크레인 인형뽑기 게임(2019 카카오)

 

https://school.programmers.co.kr/learn/courses/30/lessons/64061

프로그래머스

---

프로그래머스 레벨 1

스택을 활용하면 간단하게 해결할 수 있다!

 

나의 풀이

#include <string>
#include <vector>
#include <stack>
#include <iostream>
using namespace std;

int solution(vector<vector<int>> board, vector<int> moves) {
    int answer = 0;
    int n = board.size();
    stack<int> st;
    for(int num: moves){
        int move = num - 1;
        int get = 0;
        for(int i=0; i<n; ++i){
            if(board[i][move] != 0){
                get = board[i][move];
                board[i][move] = 0;
                if(st.empty() || st.top() != get) st.push(get);
                else if(!st.empty() && st.top() == get) {
                    answer+=2;
                    st.pop();
                }
                break;
            }
        }
    }
    return answer;
}