[백준 15903] 카드 합체 놀이

 

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

15903번: 카드 합체 놀이

 

 

 

작은 수끼리 합하여야 최종 합이 최소가 되므로

카드 숫자가 오름차순으로 정렬되도록 하기 위하여 PriorityQueue 사용

 

 

 

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));

        String line = br.readLine();
        int cards = Integer.valueOf(line.split(" ")[0]);            //카드 갯수
        int count = Integer.valueOf(line.split(" ")[1]);            //합 횟수

        //카드 수 arr
        long[] card = new long[cards];

        line = br.readLine();
        card = Arrays.stream(line.split(" ")).mapToLong(Long::parseLong).toArray();

        PriorityQueue<Long> q = new PriorityQueue<>();
        for (int i = 0; i < card.length; i++) {
            q.add(card[i]);
        }

        Long a, b;
        for (int i = 0; i < count; i++) {
            a = q.remove();
            b = q.remove();

            a += b;
            q.add(a);
            q.add(a);
        }

        int size = q.size();
        long sum = 0;
        for (int i = 0; i < size; i++) {
            sum += q.remove();
        }

        System.out.println(sum);
    }
}

 

주의) 

합체 시 숫자가 Int 범위에 해당하는지