[백준] 15903 카드 합체 놀이

[백준] 15903 카드 합체 놀이 - Greedy / Java

graycode 2022. 11. 19. 13:42

• 문제 링크

15903번: 카드 합체 놀이

첫 번째 줄에 카드의 개수를 나타내는 수 n(2 ≤ n ≤ 1,000)과 카드 합체를 몇 번 하는지를 나타내는 수 m(0 ≤ m ≤ 15×n)이 주어진다. 두 번째 줄에 맨 처음 카드의 상태를 나타내는 n개의 자연수 a1,

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;
import java.util.StringTokenizer;

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));
        StringTokenizer st = new StringTokenizer(br.readLine());

        long n = Long.parseLong(st.nextToken());
        long m = Long.parseLong(st.nextToken());

        PriorityQueue<Long> pq = new PriorityQueue<>();

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++)
            pq.offer(Long.parseLong(st.nextToken()));

        for (int i = 0; i < m; i++) {
            long a = pq.poll();
            long b = pq.poll();

            pq.offer(a + b);
            pq.offer(a + b);
        }

        long res = 0;
        for (int i = 0; i < n; i++)
            res += pq.poll();

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

}