알고리즘/카카오

크레인 인형뽑기 게임

BSHwan 2020. 5. 17. 18:39
반응형

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

 

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

[[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]] [1,5,3,5,1,2,1,4] 4

programmers.co.kr



import java.util.ArrayList;

public class Solution {
	
	static ArrayList<Integer> bucket = new ArrayList();
	
	public static int solution(int[][] board, int[] moves) {
        int answer = 0;
        
        for(int i=0; i<moves.length; i++) {
        	int y = moves[i]-1;
        	
        	for(int j=0; j<board.length; j++) {
        		if(board[j][y] != 0) { // 크레인이 잡을 인형
        			if(!bucket.isEmpty() && bucket.get(bucket.size()-1) == board[j][y]) { // 바구니가 빈상태가 아니고 같은 인형이라면
     					bucket.remove(bucket.size()-1);
     					answer += 2;
        			} else {
        				bucket.add(board[j][y]); // 바구니에 인형을 넣는다.
        			}
        			board[j][y] = 0;
        			break;
        		}
        	}
        }        
        return answer;
    }
}
반응형