백준 1715번 카드 정렬하기 C++

출처 : https://www.acmicpc.net/problem/1715

 

작은거끼리 먼저 합칠수록 비교 횟수가 적어지기 때문에 우선순위 큐를 이용하여 제일 작은게 먼저 나오도록 해서 문제를 풀었다.

 

#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 ll = long long;

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

	priority_queue<int, vector<int>, greater<int>> pq;

	int n;
	cin >> n;
	for (int i = 0; i < n; i++) {
		int num;
		cin >> num;
		pq.push(num);
	}

	ll ans = 0;
	while (!pq.empty()) {
		ll first = pq.top();
		pq.pop();
		if (pq.empty()) break;
		ll second = pq.top();
		pq.pop();
		ll cost = first + second;
		ans += cost;
		pq.push(cost);
	}
	cout << ans;
};