【蓝桥杯ALGO-225】石子游戏 Java版

算法训练 石子游戏

时间限制：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);
    }
}