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

문제

해결 방법 

우선순위 큐(priority_queue)를 활용해서 문제 그대로 구현했음.

코드

#include <iostream>
#include <queue>
#include <vector>
using namespace std;
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    int n, size, a[2] = {}, ans = 0;
    priority_queue<int, vector<int>, greater<int>> Que;
    cin >> n;
 
    if (n == 1)
    {
        cout << "0";
        return 0;
    }
 
    for (int i = 0; i < n; ++i)
    {
        int b;
        cin >> b;
        Que.push(b);
    }
 
    while (Que.size() != 1)
    {
        ans += a[0] + a[1];
        for (int i = 0; i < 2; ++i)
        {
            a[i] = Que.top();
            Que.pop();
        }
        Que.push(a[0] + a[1]);
    }
 
    cout << ans + Que.top();
}

잡설 : 비교가 아니라 합으로 이해하고 있었어서 입력이 하나면 비교 불가라 0인걸 몰라서 왜 틀렸지 한참 해맸다....