c++ 백준_1715_"카드 정렬하기"

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

1715번: 카드 정렬하기

 

​

우선순위큐를 사용해

작은 것 두개를 뽑아서 더하고 다시 push를

반복했다.

​

카드묶음이 한개 뿐인 경우에는

비교할 게 없으므로 0 출력했다.

​

​

​

처음에 예제 입력 1을 보고

저게 70이지 왜 100이지 생각했다..

​

코딩할수록 점점 국어가 딸리는 거 같다.

 

#include <iostream>
#include <queue>
using namespace std;

int main(void) {

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

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

	int N, num;
	cin >> N;

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


	int total = 0;
	if (pq.size() == 1) {
		cout << 0;
		return 0;
	}
	while (!pq.empty()) {
		
		if (pq.size() > 2) {
			int temp = pq.top(); pq.pop();
			temp += pq.top(); pq.pop();

			total += temp;

			pq.push(temp);
		}
		else if (pq.size() == 2) {
			int temp = pq.top(); pq.pop();
			temp += pq.top(); pq.pop();

			total += temp;
		}
		else {
			total += pq.top(); pq.pop();
		}

	}
	cout << total;
}