洛谷---1090合并果子

时间:2024-04-10 21:40:22

P1090 [NOIP2004 提高组] 合并果子 / [USACO06NOV] Fence Repair G - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

  1. 使用优先队列(最小堆)来存储每种果子的数量。在 Java 中,PriorityQueue 默认是最小堆。每次取出的元素都是队列中权值最小的元素。
  2. 循环读取果子的数量并添加到优先队列中。
  3. 循环合并果子,直到队列中只剩下一个元素。每次循环中,取出最小的两个堆(即数量最少的两种果子),合并它们得到一个新的堆,并将新的堆的数量加入优先队列中。
  4. 在合并的过程中,累加总体力消耗。
package 洛谷.贪心;
import java.util.PriorityQueue;
import java.util.Scanner;
public class _1090合并果子 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        // 读取果子的种类数
        int n = scanner.nextInt();

        // 读取每种果子的数目
        // 创建一个最大堆
        // PriorityQueue<Integer> maxHeap = new PriorityQueue<>(Collections.reverseOrder());
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for (int i = 0; i < n; i++) {
            pq.add(scanner.nextInt());
        }

        // 总体力消耗
        long totalEnergy = 0;

        // 贪心合并果子
        while (pq.size() > 1) {
            int firstHeap = pq.poll();  // 取出最轻的堆
            int secondHeap = pq.poll(); // 取出次轻的堆
            int newHeap = firstHeap + secondHeap; // 合并得到新的堆
            totalEnergy += newHeap; // 更新总体力消耗
            pq.add(newHeap); // 将新的堆加入优先队列
        }

        // 输出结果
        System.out.println(totalEnergy);
    }
}