[C++] 프로그래머스 문제풀이 LEVEL 2 디펜스 게임

 

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

프로그래머스

 

 

 

#include <queue>
#include <vector>

using namespace std;

int solution(int n, int k, vector<int> enemy)
{
    int answer = 0;

    int idx = 0;
    priority_queue<int> pq;
    while(idx < enemy.size())
    {
        if(n < enemy[idx] && k <= 0) break;
        n -= enemy[idx];
        pq.push(enemy[idx]);
        if(n < 0)
        {
            n += pq.top();
            pq.pop();
            --k
        }
        ++idx;
    }
    answer = idx;
    
    return answer;
}