洛谷P1090 Java解法

题目出处点这里
洛谷P1090 Java解法_第1张图片思路:先将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);
	}
}

你可能感兴趣的:(洛谷)