[JAVA] [그리디] Pro 디펜스 게임

문제

https://school.programmers.co.kr/learn/courses/30/lessons/142085?language=java 

프로그래머스

코드

import java.util.*;

class Solution {
    public int solution(int n, int k, int[] enemy) {
        // 무적권의 수가 적 배열 길이보다 클 때
        if(k>enemy.length){
            return enemy.length;
        }
        
        int answer = k-1;
        PriorityQueue<Integer> pq=new PriorityQueue<>();
        // 무적권의 길이만큼은 무조건 막을 수 있으므로
        for(int i=0; i<k; i++){
            pq.add(enemy[i]);
        }
        
        for(int i=k; i<enemy.length; i++){
            // 한개씩 추가하면서
            pq.add(enemy[i]);
            // 가장 작은 값을 제거해주고
            n-=pq.poll();
            // 남아있는 병사 수가 0보다 작아지면 break
            if(n<0){
                break;
            }
            answer=i;
        }
        
        return answer+1;
    }
}