问题描述
石子游戏的规则如下:
地上有n堆石子,每次操作可选取两堆石子(石子个数分别为x和y)并将它们合并,操作的得分记为(x+1)×(y+1),对地上的石子堆进行操作直到只剩下一堆石子时停止游戏。
请问在整个游戏过程中操作的总得分的最大值是多少?
输入格式
输入数据的第一行为整数n,表示地上的石子堆数;第二行至第n+1行是每堆石子的个数。
输出格式
程序输出一行,为游戏总得分的最大值。
样例输入
10
5105
19400
27309
19892
27814
25129
19272
12517
25419
4053
样例输出
15212676150
这道题思路比较简单,对于输入的数进行排序,从最高值开始,与前一个石子树相乘获得当前步骤的分数,同时要记得更新新的合并的石子推的数量。主要计算步骤如下两行代码:
result+=(stone[i]+1)*(stone[i-1]+1);
stone[i-1]=stone[i]+stone[i-1];
完整代码如下
import java.util.*;
public class 石子游戏 {
public static void main(String[] args)
{
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
long result=0;
long[] stone=new long[n];
for(int i=0;i<n;i++){
stone[i]=sc.nextInt();
}
Arrays.sort(stone);
for(int i=stone.length-1;i>0;i--)
{
result+=(stone[i]+1)*(stone[i-1]+1);
stone[i-1]=stone[i]+stone[i-1];
}
System.out.print(result);
}
}