[Programmers] 2019 카카오 개발자 겨울 인턴십 - 크레인 인형뽑기 게임

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

 

문제 풀이:

stack을 이용하여 쌓아 올리면 됨.

 

소스 코드:

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

int cur[32];
stack<int> st;
int solution(vector<vector<int>> board, vector<int> moves) {
	int answer = 0;
	int row = board.size();
	int col = board[0].size();

	//init
	for (int i = 0; i < col; i++) {
		for (int j = 0; j <row; j++) {
			if (board[j][i] != 0) {
				cur[i] = j;
				break;
			}
		}
	}
	for (int i = 0; i < moves.size(); i++) {
		int curP = moves[i]-1;
		if (cur[curP] >=row) continue;
		if (st.empty()) {
			st.push(board[cur[curP]][curP]);
		}
		else {
			if (st.top() == board[cur[curP]][curP]) {
				answer += 2;
				st.pop();
			}
			else st.push(board[cur[curP]][curP]);
		}
		cur[curP]++;
	}

	return answer;
}