赞
踩
P1090 [NOIP2004 提高组] 合并果子 / [USACO06NOV] Fence Repair G - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
- 使用优先队列(最小堆)来存储每种果子的数量。在 Java 中,
PriorityQueue
默认是最小堆。每次取出的元素都是队列中权值最小的元素。- 循环读取果子的数量并添加到优先队列中。
- 循环合并果子,直到队列中只剩下一个元素。每次循环中,取出最小的两个堆(即数量最少的两种果子),合并它们得到一个新的堆,并将新的堆的数量加入优先队列中。
- 在合并的过程中,累加总体力消耗。
- 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);
- }
- }
-
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。