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