본문 바로가기
백준/1 - 5000

[백준] 1715번 : 카드 정렬하기(JAVA)

by lms0806 2021. 10. 29.
728x90
반응형

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

 

1715번: 카드 정렬하기

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

www.acmicpc.net

풀이

우선순위 큐를 이용하여 가장 작은 수 더한값을 answer에 더해주고 더한값을 우선순위큐에 넣는 방식으로 pq의 크기가 1이 될때까지 반복하시면 됩니다.

 

소스코드

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		PriorityQueue<Integer> pq = new PriorityQueue<>();
		
		int n = Integer.parseInt(br.readLine());
		
		while(n --> 0) {
			pq.add(Integer.parseInt(br.readLine()));
		}
		
		long answer = 0;
		while(pq.size() != 1) {
			int num = pq.poll() + pq.poll();
			
			answer += num;
			pq.add(num);
		}
		System.out.print(answer);
	}
}
728x90
반응형

댓글