[프로그래머스][JAVA]Lv. 2 - 디펜스 게임

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

프로그래머스

라운드 중에서 가능한 효율적으로 무적권을 활용하여 방어를 할 수 있는 라운드를 구하는 문제

import java.util.PriorityQueue;
class Solution {
    public int solution(int n, int k, int[] enemy) {
        int answer = 0;
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for (int i = 0; i < enemy.length; i++) {
            pq.add(enemy[i]);
            if (pq.size() > k) {
                if (!pq.isEmpty()) {
                    n -= pq.poll();
                }
            }
            if (n < 0) {
                answer = i;
                break;
            }
        }
        if (answer == 0) answer = enemy.length;
        return answer;
    }
}

 

먼저 pq에 라운드를 진행할 때마다 enemy배열의 내용을 넣어주었다

그러다가 k보다 pq의 사이즈가 커질 때 pq에서 가장 낮을 첫 번째 내용물을 꺼내서 n에 빼주었다

이러다가 n이 0보다 작아질때 정답을 현재 라운드(i)로 하여 정답을 출력하였다

만약 n이 0이 안되었을때는 모든 라운드를 클리어했을 테니 정답을 enemy 배열의 길이로 하여 정답을 출력하였다