Level 1. 크레인 인형뽑기 게임

Level 1. 크레인 인형뽑기 게임

 

 

큐의 배열을 이용해 인형 뽑기 머신을 클래스로 추상화시켜 해결했다.

 

어려운 문제는 아니었고 카카오에서 내는 시뮬레이션 문제 중에 재밌는 것들이 많은 것 같다.

 

#include <iostream>
#include <vector>
#include <queue>
#include <stack>

using namespace std;

class ClawMachine
{
private:
    vector<queue<int>> board;

public:
    ClawMachine(vector<vector<int>>& board) : board(vector<queue<int>>(board.size()))
    {
        for (vector<int> row : board)
            for (int i = 0; i < row.size(); ++i)
                if (row[i]) this->board[i].emplace(row[i]);
    }

    int pull(int move)
    {
        queue<int>& que = board[move - 1];

        if (que.empty())
        {
            return -1;
        }
        else
        {
            int doll = que.front(); que.pop();

            return doll;
        }
    }
};

int solution(vector<vector<int>> board, vector<int> moves)
{
    int answer = 0;
    stack<int> dolls;
    ClawMachine clawMachine(board);

    for (int move : moves)
    {
        int doll = clawMachine.pull(move);

        if (!dolls.empty() && doll == dolls.top())
            dolls.pop(), answer += 2;
        else if (doll != -1)
            dolls.emplace(doll);
    }

    return answer;
}