蓝桥杯 石子游戏 java

试题 算法训练 石子游戏

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　石子游戏的规则如下：
　　地上有n堆石子，每次操作可选取两堆石子（石子个数分别为x和y）并将它们合并，操作的得分记为(x+1)×(y+1)，对地上的石子堆进行操作直到只剩下一堆石子时停止游戏。
　　请问在整个游戏过程中操作的总得分的最大值是多少？
输入格式
　　输入数据的第一行为整数n，表示地上的石子堆数；第二行至第n+1行是每堆石子的个数。
输出格式
　　程序输出一行，为游戏总得分的最大值。
样例输入
10
5105
19400
27309
19892
27814
25129
19272
12517
25419
4053
样例输出
15212676150
数据规模和约定
　　1≤n≤1000，1≤一堆中石子数≤50000

import java.util.*;


public class 石子游戏 {


	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		long[] arr = new long[n];
		for (int i = 0; i < arr.length; i++) {
			arr[i]=sc.nextInt();
		}
		long sum = 0;
		Arrays.sort(arr);
		for (int i = arr.length-2; i >= 0; i--) {
			sum+=(arr[i+1]+1)*(arr[i]+1);
			arr[i]+=arr[i+1];
		}
		System.out.print(sum);
	}
}