NOI1995：石子合并

题目链接

[NOI1995] 石子合并

题目描述

在一个圆形操场的四周摆放 $N$ 堆石子，现要将石子有次序地合并成一堆，规定每次只能选相邻的 $2$ 堆合并成新的一堆，并将新的一堆的石子数，记为该次合并的得分。

试设计出一个算法,计算出将 $N$ 堆石子合并成 $1$ 堆的最小得分和最大得分。

输入格式

数据的第 $1$ 行是正整数 $N$，表示有 $N$ 堆石子。

第 $2$ 行有 $N$ 个整数，第 $i$ 个整数 $a_i$ 表示第 $i$ 堆石子的个数。

输出格式

输出共 $2$ 行，第 $1$ 行为最小得分，第 $2$ 行为最大得分。

样例 #1

样例输入 #1

4
4 5 9 4

样例输出 #1

43
54

提示

$1\le N\le 100$，$0\le a_i\le 20$。

算法思想

每次只能选相邻的 两堆合并成新的一堆，那么可以使用区间型动态规划的思想进行处理，下面以最小得分为例。

状态表示

$f[i][j]$表示从第$i$堆石子一直合并到第$j$堆石子的最小得分

状态计算

从最小的合并长度$2$开始计算，以每次合并为阶段，枚举合并的起点，根据最后一次合并的位置可以分为下面几种情况：

• 最后一次在$i$位置合并，即将第$i$堆石子和后面的$[i+\mathrm{1...}j]$堆石子合并，得到的分数为$f[i][i]+f[i+1][j]+s[i...j]$
• 最后一次在$i+1$位置合并，即将前面的$[i...i+1]$堆石子和后面的$[i+\mathrm{2...}j]$堆石子合并，得到的分数为$f[i][i+1]+f[i+2][j]+s[i...j]$
• …
• 最后一次在$k$位置合并，即将前面的$[i...k]$堆石子和后面的$[k+\mathrm{1...}j]$堆石子合并，得到的分数为$f[i][i+1]+f[i+2][j]+s[i...j]$
• …
• 最后一次在$j-1$位置合并，即将前面的$[i...j-1]$堆石子和第$j$堆石子合并，得到的分数为$f[i][j-1]+f[j][j]+s[i...j]$

$f[i][j]$为以上情况的最小值。其中$s[i...j]$表示本次合并得到的分数，也就是第$i$堆石子到第$j$堆石子的分数和，可以使用前缀和计算得到。

初始状态

• 为计算最小值$f[i][j]$应初始化为无穷大
• $f[i][i]$表示合并1堆，无效状态应初始化为0。

除此之外，由于是在一个圆形操场的四周摆放 $N$ 堆石子，也就是说可以从任何一点出发进行合并。因此，需要采用拆环为链的方式进行处理，最后求以任意起点开始求合并分数的最小值。

时间复杂度

状态数为$n\times n$，状态计算时需要枚举最后一次合并位置，因此时间复杂度为$O(n^3)$。

代码实现

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 110 * 2, INF = 0x3f3f3f3f;
int f[N][N], g[N][N];
int a[N], s[N];

int main()
{
    int n;
    cin >> n;
    for(int i = 1; i <= n; i ++) 
    {
        cin >> a[i];
        a[n + i] = a[i]; //拆环为链
    }
    //计算前缀和
    for(int i = 1; i < 2 * n; i ++) s[i] = s[i - 1] + a[i];
    //枚举每次合并的长度，最小长度为2    
    for(int len = 2; len <= n; len ++)
        for(int i = 1; i + len - 1 < 2 * n; i ++) //枚举开始合并的位置
        {
            int j = i + len - 1; //合并结束的位置
            f[i][j] = INF; //初始状态
            //枚举最后一次合并的位置k
            for(int k = i; k < j; k ++)
            {
                f[i][j] = min(f[i][j], f[i][k] + f[k + 1][j] + s[j] - s[i - 1]); //最小得分
                g[i][j] = max(g[i][j], g[i][k] + g[k + 1][j] + s[j] - s[i - 1]); //最大得分
            }
        }
    
    int minx = 1e9, maxx = 0;
    for(int i = 1; i <= n; i ++) //枚举合并起点
    {
        minx = min(minx, f[i][i + n - 1]);
        maxx = max(maxx, g[i][i + n - 1]);
    }
    cout << minx << '\n' << maxx;
    
    return 0;
}