Java 使用堆排序的一个小例题

CCPC-Wannafly Winter Camp Day5 (Div2的C题)

题目描述

你有一个数列a₁，a₂，a₃，…a_n 。你可以进行这样的一次操作，每次选择数列中其中一个数然后将其除2下取整，也就是选择一个数a_i，变成⌊a_i/2⌋。

请问对这些数字进行不超过k次操作，这些数字的总和最小值可能是多少。

输入描述

第一行两个整数n，k。(1 ≤ n ≤ 10⁵，0 ≤ k ≤ 10⁹)，表示数字个数和操作个数。
接下来n行，一共n个整数a₁，a₂，a₃，…a_n，保证 1 <= a_i <= 10⁹。

输出描述

一个数字表示答案

样例输入

5 3
2 4 7 9 7

样例输出

16

因为要求数字总和最小，所以使用堆排序（构造大顶堆）每次弹出最大的再除2后放入堆中。最后把堆中所有值加起来即为数字最小总和。
作者小白一枚，使用Java，代码如下。

import java.util.Scanner;
import java.util.PriorityQueue;

public class Main{
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);

        //输入数字的数量
        int sum = in.nextInt();
        //进行操作的数量
        int coopNum = in.nextInt();
        //用来记录结果
        long result = 0;
        //构造优先队列来存储数据
        PriorityQueue <Integer> pq = new PriorityQueue<>(sum, (Integer a, Integer b) -> {
            return b - a;
        });
		//初始添加数据
        for (int i = 0; i < sum; i++) {
            pq.add(in.nextInt());
        }
        //弹出最大的除2再放入
        for (int i = 0; i < coopNum; i++) {
            pq.add(pq.remove() / 2);
        }
        while (!pq.isEmpty()){
            result += pq.remove();
        }
        System.out.println(result);
    }
}

小白一枚，欢迎各位批评指正。