programmers_디펜스 게임_java

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

프로그래머스

import java.util.*;

class Solution {
    public int solution(int n, int k, int[] enemy) {
        int answer = 0;
        
        int enemySize = enemy.length;
        PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2) -> o2 - o1); // 내림차순
        
        for(int i = 0; i<enemySize; i++){
            pq.offer(enemy[i]);
            n -= enemy[i];
            if(n < 0){ // 병사가 적의 수보다 적어진 경우
                if(k>0){ // 무적권이 남아 있는 경우 사용
                    n+=pq.poll(); // 현재까지의 값중 가장 큰 값에 무적권 사용
                    k--;
                }else{
                    return i; // 더이상 라운드를 진행 할 수 없는 경우
                }
            }
        }
        
        answer = enemySize; // 모든 라운드를 진행한 경우
        return answer;
    }
}