백준 30892번 상어 키우기 C++, 우선순위 큐, 최소 힙, 최대 힙

리뷰

최소 힙과 최대 힙을 사용하여 문제를 풀이하였다.

처음엔 스택을 사용하려 했으나 인입되는 순서 관리 골치아파서 힙으로 풀릴 것 같아 힙을 사용하였다.

 

문제 풀이

1. n, k, t 값을 받아와 주고 최소힙과 최대힙을 초기화 해준다.
2. t보다 크거나 같은 값은 bigger로, 작은 값은 lower 힙으로 보내준다.
3. cnt가 k보다 작을 경우 while루프를 돌려 lower에서 가장 큰 값을 t에 더해주고 해당 요소를 빼준다.
4. 요소를 빼주고 나서는 변경된 t값에 맞추어 최소힙과 최대힙을 최신화 해줘야 한다.

 

참고 사항

lower에서 가장 큰 값을 빼주기 전에 while문을 통해 bigger에 존재하는 t보다 작은 값들을 lower로 옮겨 주었다.

 

 

정답 코드

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

int n, k;
long long t;

int main() {
	cin >> n >> k >> t;
	int cnt = 0;
	vector<int> lst(n);
	for (int i = 0; i < n; i++) {
		cin >> lst[i];
	}
	priority_queue<int> lower;
	priority_queue<int, vector<int>, greater<int>> bigger;

	while (!lst.empty()) {
		if (lst.back() >= t) {
			bigger.push(lst.back());
			lst.pop_back();
		}
		else {
			lower.push(lst.back());
			lst.pop_back();
		}
	}

	while (cnt < k) {
		while (!bigger.empty() && bigger.top() < t) {
			lower.push(bigger.top());
			bigger.pop();
		}
		if (lower.empty()) break;
		t += lower.top();
		lower.pop();
		cnt++;
	}
	cout << t;
}