[NOIP 2004 提高组] 合并果子 Java

import java.util.*;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		PriorityQueue<Long> q = new PriorityQueue<>(); // 优先队列自动排序
		for (int i = 0; i < n; i++) {
			q.add(sc.nextLong());
		}
		long total = 0;
		while (q.size() >= 2) {
			long sum = q.poll() + q.poll(); // 取出头部最小的两个数
			total += sum;
			q.add(sum); // 合并后放回队列
		}
		System.out.println(total);
	}
}

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

### NOIP 2004 提高 合并果子 Python 解题思路 #### 背景描述 合并果子问题是经典的贪心算法题目之一。给定若干堆果子的数量,每次可以选取两堆数量最少的果子将其合并成一堆,并记录此次合并所花费的成本(即这两堆果子数之和)。最终目标是最小化总成本。 #### 思路分析 为了最小化合并过程中的总成本,应该优先考虑将较小的两堆先合并起来。这样做的好处是可以减少后续较大规模合并时所需付出的成本。具体来说: - 使用一个小根堆来存储每堆果子的数量。 - 每次取出两个最小值进行合并操作,并把新得到的结果重新放回堆中继续参与下一轮比较。 - 记录每一次合并产生的费用直到只剩下一堆为止[^1]。 #### 实现方法 基于上述策略,在Python编程语言环境下可以通过`heapq`模块轻松构建这样一个高效的小顶堆结构来进行求解。 ```python import heapq def min_cost_to_merge_fruits(fruit_piles): # 将所有的果子堆加入到一个列表里, 并转换为最小堆 heapq.heapify(fruit_piles) total_cost = 0 while len(fruit_piles) > 1: # 取出当前最小的两堆果子 first_min = heapq.heappop(fruit_piles) second_min = heapq.heappop(fruit_piles) current_cost = first_min + second_min # 更新总的消耗代价 total_cost += current_cost # 把这次合并后的结果再加回到堆里面去 heapq.heappush(fruit_piles, current_cost ) return total_cost ``` 此函数接收一个整型数作为输入参数,代表初始状态下各堆果子的具体数目;返回的是完成全部合并不需要额外空间复杂度下的最低可能耗费时间/次数。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值