[JAVA] 디펜스 게임 - PriorityQueue

문제 설명
준호는 요즘 디펜스 게임에 푹 빠져 있습니다. 디펜스 게임은 준호가 보유한 병사 n명으로 연속되는 적의 공격을 순서대로 막는 게임입니다. 디펜스 게임은 다음과 같은 규칙으로 진행됩니다.

준호는 처음에 병사 n명을 가지고 있습니다.
매 라운드마다 enemy[i]마리의 적이 등장합니다.
남은 병사 중 enemy[i]명 만큼 소모하여 enemy[i]마리의 적을 막을 수 있습니다.
예를 들어 남은 병사가 7명이고, 적의 수가 2마리인 경우, 현재 라운드를 막으면 7 - 2 = 5명의 병사가 남습니다.
남은 병사의 수보다 현재 라운드의 적의 수가 더 많으면 게임이 종료됩니다.
게임에는 무적권이라는 스킬이 있으며, 무적권을 사용하면 병사의 소모없이 한 라운드의 공격을 막을 수 있습니다.
무적권은 최대 k번 사용할 수 있습니다.
준호는 무적권을 적절한 시기에 사용하여 최대한 많은 라운드를 진행하고 싶습니다.

준호가 처음 가지고 있는 병사의 수 n, 사용 가능한 무적권의 횟수 k, 매 라운드마다 공격해오는 적의 수가 순서대로 담긴 정수 배열 enemy가 매개변수로 주어집니다. 준호가 몇 라운드까지 막을 수 있는지 return 하도록 solution 함수를 완성해주세요.

제한사항
1 ≤ n ≤ 1,000,000,000
1 ≤ k ≤ 500,000
1 ≤ enemy의 길이 ≤ 1,000,000
1 ≤ enemy[i] ≤ 1,000,000
enemy[i]에는 i + 1 라운드에서 공격해오는 적의 수가 담겨있습니다.
모든 라운드를 막을 수 있는 경우에는 enemy[i]의 길이를 return 해주세요.

해당 문제를 정렬이나 map을 통해 하려고 시도하였으나
적절하지 않았고 다른 풀이를 참고하다가 PriorityQueue를 사용한 것을 보고
바로 다시 접근해 풀이를 시도하였다

풀이

import java.util.*;
class Solution {
    public int solution(int n, int k, int[] enemy) {
        int answer = 0;
        int size = enemy.length;
        // 만약 라운드보다 무적권이 더 많을 경우 해당 라운드 전체 수 반환
        if(k>=size) return size;
        
        // priorityQueue에는 무적권을 사용하여 처리한 적의 수를 저장
        PriorityQueue<Integer> que = new PriorityQueue<>();

        for(int i=0; i<size; i++){
            // 무적권 처리적 넣기
            if(que.size()<k){
                que.offer(enemy[i]);
            // 무적권 사용 횟수를 초과하면 이전값과 비교
            } else{
                // 만약 적의 수가 더 많으면
                if(que.peek()<enemy[i]){
                    que.offer(enemy[i]);
                    n -= que.poll();
                    // 병사가 있다면 더하고 없으면 종료
                    if(n>=0) answer++;
                    else break;
                } else {
                    n-= enemy[i];
                    // 병사가 있다면 더하고 없으면 종료
                    if(n>=0) answer++;
                    else break;
                }
            }       
        }
        return answer+k; // 무적권 수와 버튼 라운드수의 합
    }
}