【洛谷】P1880 石子合并

题目地址：

https://www.luogu.com.cn/problem/P1880

题目描述：
在一个圆形操场的四周摆放$N$堆石子，现要将石子有次序地合并成一堆，规定每次只能选相邻的$2$堆合并成新的一堆，并将新的一堆的石子数，记为该次合并的得分。试设计出一个算法,计算出将$N$堆石子合并成$1$堆的最小得分和最大得分。

输入格式：
数据的第$1$行是正整数$N$，表示有$N$堆石子。第$2$行有$N$个整数，第$i$个整数$a_i$表示第$i$堆石子的个数。

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

数据范围：
$1\le N\le 100$，$0\le a_i\le 20$。

由于是环形摆放的石子，可以开一个新数组$b$，使得$b[i]=b[i+n]=a[i],1\le i\le n$。设$f[l][r]$是考虑$b[l:r]$的情况下能得到的最小得分，那么我们要求的就是${\min }_{1\le i\le n}f[i][i+n-1]$。如果$l=r$，则无法再合并，此时$f[l][r]=0$；否则的话，可以根据最后一次合并的间隔点来分类，所以有： f [ l ] [ r ] = min ⁡ l ≤ k ≤ r − 1 { f [ l ] [ k ] + f [ k + 1 ] [ r ] + ∑ l ≤ i ≤ r b [ i ] } f[l][r]=\min_{l\le k\le r-1} \{f[l][k]+f[k+1][r]+\sum_{l\le i\le r} b[i]\} f[l][r]=l≤k≤r−1min​{f[l][k]+f[k+1][r]+l≤i≤r∑​b[i]}可以用记忆化搜索来做。由于要频繁计算区间和，可以预处理一下前缀和。代码如下：

#include <iostream>
#include <cstring>
#define x first
#define y second
using namespace std;
using PII = pair<int, int>;

const int N = 110 << 1, INF = 1e9;
int n, a[N], sum[N];
int f[N][N], g[N][N];

PII dfs(int l, int r) {
  if (~f[l][r]) return {f[l][r], g[l][r]};
  if (l >= r) {
    f[l][r] = g[l][r] = 0;
    return {0, 0};
  }

  int x = INF, y = -INF;
  for (int k = l; k < r; k++) {
    auto le = dfs(l, k), ri = dfs(k + 1, r);
    int s = sum[r] - sum[l - 1];
    x = min(x, le.x + ri.x + s);
    y = max(y, le.y + ri.y + s);
  }

  f[l][r] = x, g[l][r] = y;
  return {x, y};
}

int main() {
  memset(f, -1, sizeof f);
  memset(g, -1, sizeof g);

  scanf("%d", &n);
  for (int i = 1; i <= n; i++) {
    scanf("%d", &a[i]);
    a[i + n] = a[i];
  }

  for (int i = 1; i <= n << 1; i++) sum[i] = sum[i - 1] + a[i];

  int x = INF, y = -INF;
  for (int i = 1; i <= n; i++) {
    auto p = dfs(i, i + n - 1);
    x = min(x, p.x);
    y = max(y, p.y);
  }

  printf("%d\n%d\n", x, y);
}

时间复杂度$O(n^3)$，空间$O(n^2)$。