今天再讲一篇关于利用贪心算法解决的题目。
一、金条切割问题
1、题目描述
一块金条切成两半,是需要花费和长度数值一样的铜板的。比如长度为20的金条,不管怎么切,都要花费20个铜板。一群人想整分整块金条,怎么分最省铜板?输入一个数组,返回分割的最小代价。
例如:
给定数组{10,20,30},代表一共三个人, 整块金条长度为10 + 20 + 30 = 60,金条要分成10, 20, 30三个部分(不考虑顺序)。
如果先把长度60的金条分成10和50,花费60;再把长度50的金条分成20和30,花费50;一共花费110铜板。
但如果先把长度60的金条分成30和30,花费60;再把长度30金条分成10和20,花费30;一共花费90铜板。
2、思路
(1)准备一个小根堆。将数组放到这个小根堆里。
(2)每次弹出堆顶的两个数求和为A,将A再放回小根堆里。
(3)一直执行第2步,直到堆只剩一个数。最后,每一次第二步A的累加和即是最后的结果。
例如给定的金条长度为150,要分成10、20、30、40、50的块,最后花费的铜板数量即是上图中蓝色圆圈的和,即150+60+90+30=330。
也就是我们代码求解的时候是从叶子往根求的,求完后再从根往叶子即是金条的切割顺序,最后所有的叶子即是需要切成的块的大小。
3、代码
/**
* @author Java和算法学习:周一
*/
public static int lessMoneySplitGold(int[] arr) {
if (arr == null || arr.length == 0) {
return 0;
}
// 准备一个小根堆
Queue<Integer> queue = new PriorityQueue<>();
// 将所有数放到小根堆中
for (Integer s : arr) {
queue.offer(s);
}
int result = 0;
int current;
while (queue.size() > 1) {
// 每次弹出堆顶两个数求和
current = queue