[JAVA] 프로그래머스 디펜스 게임_우선순위 큐

 

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

프로그래머스

 

 

접근 : 

 

  - 무적권의 갯수가 적의 수 이상인 경우에는 모든 라운드를 클리어 할 수 있으므로 라운드 수를 반환

 

  - 적을 앞 라운드 부터 순서대로 확인하여 n을 감소시켜 준다 만약 n 이 음수가 되는 경우에는 무적권을 사용하고 이전에서 가장 큰 수를 다시 n 에 더해준다.

 

  - 무적권을 모두 소모할 때까지 반복하고 만약 무적권을 모두 소진한다면 해당라운드는 클리어하지 못하였으므로 ans - 1반환해준다.

 

 

 

 

코드 구현 : 

 

import java.util.*;

class Solution {
    public int solution(int n, int k, int[] enemy) {
        if (enemy.length <= k) return enemy.length;
        int answer = 0;
        PriorityQueue<Integer> queue = new PriorityQueue<>(Collections.reverseOrder());

        for (int e : enemy) {
            queue.add(e);
            answer++;
            n -= e;
            if(n >= 0) continue;
            if(k == 0) return answer-1;

            n = n + queue.poll();
            k--;
        }
        return enemy.length;
    }
}