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

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

1715번: 카드 정렬하기

 

저는 해당 문제를 우선순위 큐를 이용해 문제를 해결하였습니다.

우선 순위 큐를 통해 작은 값 두개를 꺼내서 더해주고 이 값을 result에 더하고 다시 큐에 넣었습니다.

위의 과정을 큐의 사이즈가 1이 될때까지 하였습니다.

 

package BOJ.ETC;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.PriorityQueue;
import java.util.Queue;
import java.util.StringTokenizer;
import java.util.function.Function;

public class BOJ_1715 {

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Function<String,Integer> stoi = Integer::parseInt;
        int n = stoi.apply(br.readLine());
        Queue<Integer> q = new PriorityQueue<>();
        for(int i = 0 ; i < n ; i++){
            int num = stoi.apply(br.readLine());
            q.offer(num);
        }
        long result = 0;
        while(q.size() > 1){
            int num1 = q.poll();
            int num2 = q.poll();
            q.offer(num1 + num2);
            result += (num1+num2);
        }
        System.out.println(result);
    }
}