问题描述
石子游戏的规则如下:
地上有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.Scanner;
public class Main{
/*
* 类似哈夫曼树,不过本题是每次选取最大的两个数
* */
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
long[] a = new long[n];
for(int i=0;i<n;i++){
a[i] = sc.nextInt();
}
boolean flag=false;
for(int i=n-1;i>=0;i--){//冒泡排序(递减)
flag = false;
for(int j=0;j<i;j++){
if(a[j]<a[j+1]){
long temp = a[j];
a[j] = a[j+1];
a[j+1] = temp;
flag = true;
}
}
if(!flag)//没有交换,排序结束
break;
}//for
long result = 0;
for(int i=0;i<n-1;i++){
result += (a[i]+1)*(a[i+1]+1);
a[i+1] = a[i]+a[i+1];
}
System.out.println(result);
}
}