본문 바로가기

Algorithm/Greedy

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

반응형

www.acmicpc.net/problem/1715

 

1715번: 카드 정렬하기

정렬된 두 묶음의 숫자 카드가 있다고 하자. 각 묶음의 카드의 수를 A, B라 하면 보통 두 묶음을 합쳐서 하나로 만드는 데에는 A+B 번의 비교를 해야 한다. 이를테면, 20장의 숫자 카드 묶음과 30장

www.acmicpc.net

 

min heap을 사용해 푼 문제였습니다.

 

풀이방법

1. 카드 묶음을 입력 받고 minHeap에 push해줍니다.

2. 카드 두장을 뽑고 답 변수 ans에 더해준 다음 합쳐진 카드를 다시 minHeap에 push합니다.

 

Code

#include <bits/stdc++.h>
using namespace std;
int n, card;
int main(){
	int ans = 0;
	cin >> n;
	priority_queue<int, vector<int>, greater<int>> minHeap;

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

	for (int i = 0; i < n - 1; i++){
		card = minHeap.top();
		minHeap.pop();
		card += minHeap.top();
		minHeap.pop();
		ans += card;
		minHeap.push(card);
	}
	cout << ans;
}