蓝桥杯 算法训练 石子游戏

问题描述

　　石子游戏的规则如下：
　　地上有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);
	}
	
}