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

단순 구현 문제이다.

Stack을 이용한 단순 구현 문제이다.

 

package programmers;
 
import java.util.Stack;
 
public class Solution_2019카카오겨울_크레인인형뽑기게임 {
    public static void main(String[] args) {
        
        int[][] board= {{0,0,0,0,0},{0,0,1,0,3},{0,2,5,0,1},{4,2,4,4,2},{3,5,1,3,1}};
        int[] moves= {1,5,3,5,1,2,1,4};
        
        System.out.println(solution(board,moves));
        
    }
    
    static int solution(int[][] board,int[] moves) {
        int r=board.length;
        int answer=0;
        Stack<Integer> stack = new Stack<>();
 
        for(int move : moves) {
            int index=move-1;
            
            for(int i=0;i<r;i++) {    //행의 처음부터 확인
                if(board[i][index]!=0) {
                    int num=board[i][index];
                    board[i][index]=0;
                    
                    if(stack.size()!=0) {
                        if(stack.peek()==num) {
                            stack.pop();
                            answer=answer+2;
                        }
                        else {
                            stack.push(num);
                        }
                    }else {
                        stack.push(num);
                    }
                    
                    break;
                }
            }
        }
        return answer;
    }
}