99클럽 코테 스터디 38일차 TIL + 오늘의 학습 키워드

☑️ 문제:  디펜스 게임

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

프로그래머스

 

☑️ Code

import java.util.PriorityQueue;
import java.util.Collections;

class Solution {

    public int solution(int n, int k, int[] enemy) {
        PriorityQueue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());

        int answer = 0;
        for (int e : enemy) {
            pq.offer(e);
            answer++;
            n -= e;
            if (n < 0) {
                if(k==0){
                    return answer - 1;
                }
                n = n + pq.poll();
                k--;
            }
        }

        return enemy.length;
    }
}