题目描述:
小明数学不太好,如果他计算a+b需要花费(a+b)的时间,现在有n个数字,让小明去求这n个数字的和,求小明计算这n个数字所花费的最小时间;
备注:每次都要先算最小的两个数的和,这样总时间才会最小。
输入:[1,2,3]
输出:9
说明:
1+2需要3秒
3+3需要6秒
总共需要9秒
输入:[1,2,3,4,5]
输出:33
说明:
1+2需要3秒
3+3需要6秒
4+5需要9秒
6+9需要15秒
可以使用优先队列PriorityQueue实现;
public static void main(String[] args) {
PriorityQueue<Integer> queue = new PriorityQueue<>();
queue.add(1);
queue.add(2);
queue.add(3);
queue.add(4);
queue.add(5);
int sum = 0;
while (queue.size() >= 2) {
int a = queue.poll();
int b = queue.poll();
int num = a + b;
sum += num;
queue.add(num);
}
System.out.println(sum);
}