蓝桥杯题解 谈判 Java答案

原题链接
题目描述
在很久很久以前，有 个部落居住在平原上，依次编号为 到 。第 个部落的人数为 。
有一年发生了灾荒。年轻的政治家小蓝想要说服所有部落一同应对灾荒，他能通过谈判来说服部落进行联合。
每次谈判，小蓝只能邀请两个部落参加，花费的金币数量为两个部落的人数之和，谈判的效果是两个部落联合成一个部落（人数为原来两个部落的
人数之和）。
输入描述
输入的第一行包含一个整数 ，表示部落的数量。
第二行包含 个正整数，依次表示每个部落的人数。
运行限制
最大运行时间：1s
最大运行内存: 128M

package testtt;

import java.math.BigDecimal;
import java.math.MathContext;
import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Scanner;
import java.util.TreeSet;


public class Main {
	
	
	public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        TreeSet<Integer> set = new TreeSet<>(new Comparator<Integer>() {

			@Override
			public int compare(Integer o1, Integer o2) {
				return o1 - o2 == 0? 1: o1 - o2;
				
			}
		});
        int x = scan.nextInt();
        int t;
        int ans = 0;
        for(int i = 0;i < x;i ++) {
        	t = scan.nextInt();
        	set.add(t);
        }
        while(set.size() != 1) {
        	int t1 = set.pollFirst();
        	int t2 = set.pollFirst();
        	t1 += t2;
        	set.add(t1);
        	ans += t1;
       }
       System.out.println(ans);
       scan.close();
    }
}