题目:
一块金条切成两半,是需要花费和长度数值一样的铜板的。比如 长度为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铜板。 输入一个数组,返回分割的最小代价。 |
---|
可以看出,其实就是哈夫曼树,哈夫曼带权路径长度==分割铜板代价,因为哈夫曼的wpl是所有非叶子节点的加和,和这个题一样,(就是叶节点*深度的和)
构造哈夫曼用到最小堆
牛客上的有原题,但是注意牛客的测试用例范围很大,所以一定要把int改成long,我这里的int是通不过的
import java.util.PriorityQueue;
public class Trie {
public static void main(String[] args) {
PriorityQueue<Integer> heap=new PriorityQueue<Integer>();
int[] fenshu= {10,20,30,40,32};
int sum=0;
for(int i:fenshu) {
heap.add(i);
}
while(heap.size()>1) {
int temp=heap.poll()+heap.poll();
sum=sum+temp;
heap.add(temp);
}
System.out.println(sum);
}
}
补充,如果java的优先队列,如果想用大根堆,只需要传一个比较器就可以了,参考这篇博文 java中的大根堆和小根堆