[Programmers] 크레인 인형뽑기 게임

#include <string> #include <vector> #include <stack> #include <iostream> using namespace std; int solution(vector<vector<int>> board, vector<int> moves) { int answer = 0; //시뮬레이션 //같은 숫자가 2개 쌓이면 없어짐 stack<int> st; int size = board[0].size(); //인형이 없더라도 0으로 차있기 때문에 사이즈는 같음 //중간에 0인 부분은 밑으로 끌어내려야 한다. // for (int i = 0; i < board.size(); i++) { // int idx = 0; // while(idx < size) { // if (!board[i][idx]) { // while (idx > 0 && !board[i][idx] && board[i][idx-1]) { // swap(board[i][idx], board[i][idx-1]); // idx--; // } // } // idx++; // } // } // for (int i = 0; i < board.size(); i++) { // for (int j = 0; j < size; j++) { // cout << board[i][j] << ' '; // } // cout << '\n'; // } for (int i = 0; i < moves.size(); i++) { int idx = moves[i]-1; int tmpS =0; while(tmpS < size ) { if (board[tmpS][idx]) break; tmpS++; } if (tmpS == size) continue; cout << idx << " " << tmpS << " " << board[idx][tmpS] << '\n'; if (st.empty()) { st.push(board[tmpS][idx]); } else{ if (st.top() == board[tmpS][idx]) { answer += 2; st.pop(); } else st.push(board[tmpS][idx]); } board[tmpS][idx] = 0; } // for (int i = 0; i < board.size(); i++) { // for (int j = 0; j < size; j++) { // cout << board[i][j] << ' '; // } // cout << '\n'; // } return answer; }