C++ (코딩테스트) - 디펜스 게임(프로그래머스)

문제

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

프로그래머스

 

내가 쓴 풀이

#include <string>
#include <vector>
#include <queue>

using namespace std;

int solution(int n, int k, vector<int> enemy) {
	priority_queue<int> pq;
	pq.empty();

	for (int i = 0; i < enemy.size(); i++)
	{
		pq.push(enemy[i]);
		n -= enemy[i];
		if (n < 0)
		{
			if (k-- <= 0) { return i; }
			n += pq.top();
			pq.pop();
		}
	}
	return enemy.size();
}

 

 

참고자료

(우선순위큐)

https://jungeu1509.github.io/algorithm/use-priorityqueue/