题目出处点这里
思路:先将n个果堆根据重量从小到大,每次先取最小的两个求重量和并把这两个果堆从原来n个果堆中去掉,再把这个重量和放入原先堆再进行从小到大的排序,重复这个过程并把每次的重量和相加就可以得到结果
注意:当果堆数为n时,需要进行的排序次数为n-1;当n比较大时,进行n-1次排序显然会TLE,因此我们用小顶堆,也就是优先队列就可以解决这一问题,减少运行时间。
代码如下:
package greedy;
import java.util.PriorityQueue;
import java.util.Scanner;
public class P1090 {
static Integer head;
static Integer next;
static Integer newNum;
static long sum;
static PriorityQueue<Integer> list = new PriorityQueue<Integer>();
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
for (int i = 0; i < n; i++) {
list.add(sc.nextInt());
}
for (int i = 0; i < n - 1; i++) {
head = list.poll();
next = list.poll();
newNum = head + next;
list.add(newNum);
sum = sum + newNum;
}
System.out.println(sum);
}
}