问题描述
石子游戏的规则如下:
地上有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
解题思路:
题意有点难懂,容易理解偏。题意是:在一堆石子中,两两合并得到的第三堆继续加入石子堆中进行合并。同时将合并的两堆分别加一再相乘得到乘积,每次合并都将乘积累加起来,求累加的最大和。要想累加和最大,只要每次合并时都合并最大的两堆,即只要局部最优便可达到最后解最优。因为相乘时数据会很大,需要用到BigInteger类,在每次挑选时,都要进行一次排序,需要用到Comparator比较器,对大整型的比较需要比较字符串(大整型类没有对比较运算符重载),最后剩下一堆时得到的累加和即为最后答案。
java代码:
import java.io.*;
import java.math.*;
import java.util.*;
public class Main {
public static void main(String[] args) throws NumberFormatException, IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
List<BigInteger> list = new ArrayList<>();
for(int i = 0; i < n;i++) {
BigInteger big = new BigInteger(br.readLine());
list.add(big);
}
BigInteger sum = new BigInteger("0");
while(list.size()!= 1) {
Collections.sort(list,new Comparator<BigInteger>() {
@Override
public int compare(BigInteger o1, BigInteger o2) {
return o2.subtract(o1).toString().compareTo("0");
}
});
if(list.size() > 1) {
BigInteger n1 = list.remove(0);
BigInteger n2 = list.remove(0);
list.add(n1.add(n2));
BigInteger temp = n1.add(new BigInteger("1")).multiply(n2.add(new BigInteger("1")));
sum = sum.add(temp);
}
}
System.out.println(sum.toString());
}
}
提交截图: