[c++] 프로그래머스 - 크레인 인형뽑기 게임(Level 1)

문제

코딩테스트 연습 - 크레인 인형뽑기 게임

 

코드

#include <string>
#include <vector>
#include <stack>

using namespace std;

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

 

풀이(19분)

크게 복잡하지는 않았습니다. move위치의 depth를 내려가면서 인형이 있을 때 뽑은 다음 이전 stack의 top에 있을 경우 터트리는 식으로 구현했습니다.