[백준] 13975번 - 파일 합치기 3 [Java]

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

 

[백준] 13975번 - 파일 합치기 3 [Java]

[백준] 13975번 - 파일 합치기 3 [Java]

---

1.  아이디어

 

두 파일을 합치는 과정을 반복할 때 작은 두 파일을 합치는 과정을 반복해야 최소 비용으로 하나의 파일로 만들 수 있다. 이를 위해 우선순위 큐를 활용했다.

---

2. 문제풀이

 

우선순위 큐에 초기 파일들을 담은 후 두 파일을 꺼내서 합친 후 다시 우선순위 큐에 넣는 과정을 반복했다. int형 오버플로우가 발생할 수 있음에 주의해야 한다.

---

3. 코드

 

import java.io.*;
import java.util.*;

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));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;

        int T = Integer.parseInt(br.readLine());
        for (int tc = 1; tc <= T; tc++) {
            int N = Integer.parseInt(br.readLine());

            PriorityQueue<Long> pq = new PriorityQueue<>();
            st = new StringTokenizer(br.readLine());
            while (st.hasMoreTokens()) {
                pq.add(Long.parseLong(st.nextToken()));
            }

            long ans = 0;
            while (pq.size() > 1) {
                long sum = pq.remove() + pq.remove();
                pq.add(sum);
                ans += sum;
            }

            sb.append(ans).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기