题目:
一块金条切成两半,是需要花费和长度数值一样的铜板的。比如长度为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铜板。输入一个数组,返回分割的最小代价。
例如有如下数组:int[] arrForHeap={ 3, 5, 2, 7, 0, 1, 6, 4 };先把数组构造成一个小根堆,每次从小根堆里取出两个最小值,相加后再把和放回堆里,也就是通过哈夫曼的思想构造得到的二叉树,这样每次从图中的红色点向下切割分下去,得到的就是最小的代价。(28往下分割都是差不多一半一半 的分割的,所以代价最小)
public class LessMoney {
public static int lessMoney(int[] arr) {
PriorityQueue<Integer> minHeapQueue=new PriorityQueue<>(new MinheapComparator());
for(int i=0;i<arr.length;i++) {
minHeapQueue.add(arr[i]);
}
int sum=0;
int cur=0;
while(minHeapQueue.size()>1) {
cur=minHeapQueue.poll()+minHeapQueue.poll();
sum+=cur;
minHeapQueue.add(cur);
}
return sum;
}
//小根堆比较器
public static class MinheapComparator implements Comparator<Integer>{
@Override
public int compare(Integer o1, Integer o2) {
return o1-o2;
}
}
public static void main(String[] args) {
int[] arrForHeap={ 3, 5, 2, 7, 0, 1, 6, 4 };
System.out.println(lessMoney(arrForHeap));
}
}