算法训练 石子游戏
时间限制: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
分析
- 贪心算法
- 数组降序排序
Java 代码:
import java.io.*;
import java.util.*;
public class Main {
public static void main(String[] args) throws Exception {
BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(in.readLine());
// 稍后要降序排序,数组类型不能用基本类型
Long[] a = new Long[n];
for (int i = 0; i < n; i++) {
a[i] = Long.parseLong(in.readLine());
}
Arrays.sort(a, Collections.reverseOrder());
Long sum = Long.valueOf(0);
for (int i = 0; i < n - 1; i++) {
sum += (a[i] + 1) * (a[i + 1] + 1);
a[i + 1] = a[i] + a[i + 1];
}
System.out.print(sum);
}
}
FreeTechLiu
发布了11 篇原创文章 · 获赞 15 · 访问量 1968
私信
关注