[JAVA] 디펜스 게임

https://school.programmers.co.kr/learn/courses/30/lessons/142085

프로그래머스

 

 

 

 

 

import java.util.PriorityQueue;
import java.util.Collections;

class Solution {
    public int solution(int n, int k, int[] enemy) {
        int answer = 0;
        
        // 막은 적들을 내림차순으로 저장
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
        
        for(int e : enemy) {
            
            // 병사를 이용해 적을 막음
            n -= e;
            // 막은 적의 수를 저장 (내림차순으로 저장됨)
            pq.add(e);
            
            if(n < 0) {
                // 병사를 전부 소모했다면
                if(k > 0) {
                    // 무적권이 남았다면 지금까지 막은 적 중 가장 많은 수의 적을 무효화
                    int tmp = pq.poll();
                    n += tmp;
                    k--;
                }
                else {
                    // 없으면 끝난거
                    break;
                }
            }
            
            // 스테이지 클리어
            answer++;
            
        }
        
        return answer;
    }
}