99클럽 코테 스터디 38일차 디펜스 게임

📝 문제

프로그래머스

 

🔥 시도해본 접근 방식

import java.util.*;

class Solution {

    public int solution(int n, int k, int[] enemy) {
        
        int soldier = n;
        int passCard = k;
        
        PriorityQueue<Integer> queue = new PriorityQueue<>(Collections.reverseOrder());
        
        int roundIndex = 1;
        for (int round : enemy) {
            queue.add(round);
            soldier -= round;
            
            if (soldier < 0) {
                while (!queue.isEmpty()) {
                    if (passCard == 0) {
                        break;
                    }

                    soldier += queue.poll();
                    passCard--;
                    
                    if (soldier >= 0) {
                        break;
                    }
                }
            }
            
            if (soldier < 0) {
                return roundIndex - 1;
            }
            
            roundIndex++;
        }
            
        return enemy.length;
    }
}