본문 바로가기
Problem Solving/Baekjoon

[백준] 1715 카드 정렬하기 - Greedy / Java

by graycode 2022. 12. 25.

 문제 링크

 

1715번: 카드 정렬하기

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

www.acmicpc.net

 

 풀이 과정

 

 풀이 코드

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

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(br.readLine());

        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for (int i = 0; i < n; i++)
            pq.offer(Integer.parseInt(br.readLine()));

        int cnt = 0;
        while (pq.size() > 1) {
            int sum = pq.poll() + pq.poll();

            cnt += sum;
            pq.offer(sum);
        }

        bw.write(String.valueOf(cnt));
        bw.flush();
    }

}

댓글