[프로그래머스]카카오인턴십_크레딧인형뽑기

programmers.co.kr/learn/courses/30/lessons/64061

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

 

시뮬레이션 적인 문제

인형뽑기 + Stack에 대한 활용 ( 예외처리에 대한 부분 value가 0 일때 )

import java.util.Stack;
 
class Solution {
    public int pullup(int row,int N,int map[][]){
        //row번째 열에서 어떤 인형을 뽑는지 (0나오면 안뽑힌거)
        int result = 0;
        for(int i=0; i<N; i++){
            if(map[i][row] != 0 ){
                result = map[i][row];
                map[i][row] = 0 ;
                break;
            }
        }
        return result;
    }
    
    public int solution(int[][] board, int[] moves) {
        Stack<Integer> st = new Stack();
        int answer = 0;
        int N = board.length;
        for(int i=0;i<moves.length;i++){
            int value = pullup(moves[i]-1,N,board);
            if(value == 0 ) continue;
            if(!st.isEmpty()){
                int top = st.peek();
                if(top == value){
                    //같은게 2개있으면
                    answer += 2;
                    st.pop();
                }else{
                    st.push(value);
                }
            }else{st.add(value);}
            
        }        
        return answer;
    }
}