백준 23843번 콘센트 C++

https://www.acmicpc.net/problem/23843

23843번: 콘센트

 

1. 전자기기들을 내림차순으로 정렬

2. 작은 수가 제일 먼저 뽑히게 하는 우선순위 큐 생성

3. 앞에서 m개만큼 전자기기를 우선순위 큐에 삽입

4. m ~ n-1까지 전자기기들의 시간을 우선순위 큐의 top을 뽑은 값에 더하고 다시 우선순위 큐에 삽입

    ( 일이 가장 빨리 끝난 플러그에 바로 일 할당)

5. 우선순위 큐를 pop하면서 가장 마지막에 남은 원소가 답이다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n, m;
vector<int> devices;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> n >> m;
	devices = vector<int>(n);
	for (int i = 0; i < n; i++) {
		cin >> devices[i];
	}
	sort(devices.begin(), devices.end(), greater<int>());
	priority_queue<int, vector<int>, greater<int>> pq;
	for (int i = 0; i < m; i++) {
		if (i >= n) break;
		pq.push(devices[i]);
	}
	for (int i = m; i < n; i++) {
		int pop = pq.top();
		pq.pop();
		pop += devices[i];
		pq.push(pop);
	}
	for (int i = 0; i < m - 1; i++) {
		pq.pop();
	}
	cout << pq.top();
};