크레인 인형뽑기 게임

[문제]

 

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

 

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

 

[해결]

 

인형이 있는 board는 queue배열로 만들어서 

moves에 index로 바로 가장 위에 있는 인형을 뽑는다.

 

바구니는 stack을 사용하여 인형 넣고, 빼는 순서의 고려를 줄인다.

 

 [코드]

 

class kakaoSolution {
	public int solution(int[][] board, int[] moves) {
		int answer = 0;
		int size = board[0].length;
		Queue<Integer>[] queue = new Queue[size];
		for (int i = 0; i < size; i++) {
			queue[i] = new LinkedList<Integer>();
		}

		for (int i = 0; i < size; i++) {
			for (int j = 0; j < size; j++) {
				if (board[i][j] == 0)
					continue;

				queue[j].offer(board[i][j]);
			}
		}
		 
		// 바구니 만들기
		Stack<Integer> bucket = new Stack<Integer>();
		int num;
		// 바구니에 있는 거 뺴기
		for (int i = 0; i < moves.length; i++) {
			if (queue[moves[i] - 1].size()==0)	continue;
			
			num = queue[moves[i]-1].poll();
			if (i == 0) {
				bucket.push(num);
				continue;
			}
			
			if(!bucket.isEmpty()) {
				int bucketTopnum = bucket.peek(); 
				if(num == bucketTopnum) {
					bucket.pop();
					answer= answer+2;
					continue;
				}
			}	
				bucket.push(num);	
		}
		return answer;
	}
}