[BAEKJOON] 1715. 카드 정렬하기

 

문제링크

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

1715번: 카드 정렬하기

 

PriorityQueue 를 사용한다면 쉽게 해결할 수 있는 문제

100%에서 틀린다면 묶음이 하나 있는 경우 => 비교횟수 = 0

나는 묶음이 하나 있다면 그 묶음의 수를 return 해서 틀렸다.

package acmicpc.Gold4;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;

public class 카드_정렬하기 {
	// 가장 작은 수의 묶음들을 먼저 비교하면 된다.
	
	static int n, ans;
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		n = Integer.parseInt(br.readLine());
		
		PriorityQueue<Integer> pq = new PriorityQueue<>();
		for(int i = 0 ; i < n ; i++) {
			pq.add(Integer.parseInt(br.readLine()));
		}
		ans = 0;
		
		while(pq.size() != 1) {
			int a = pq.poll();
			int b = pq.poll();
			
			ans += (a + b);
			pq.add(a + b);
		}
		
		System.out.println(ans);
		
	}
}