石子合并
题目描述
题目描述
在一个圆形操场的四周摆放 N 堆石子,现要将石子有次序地合并成一堆.规定每次只能选相邻的2堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。
试设计出一个算法,计算出将 N 堆石子合并成 1堆的最小得分和最大得分。
题目链接
思路
如果是普通的区间动态规划,我们可以直接从2到n枚举每一个区间的长度,确定区间的起始点和终点,然后再枚举分点,通过前缀和计算每次费用增量就可以。代码如下
for (int len = 2; len <= n; ++len) {
for (int i = 1; i <= n - len + 1; ++i) {
int j = i + len - 1;
for (int k = i; k < j; ++k) {
dp1[i][j] = min(dp1[i][j], dp1[i][k] + dp1[k + 1][j] + sum[j] - sum[i - 1]);
dp2[i][j] = max(dp2[i][j], dp2[i][k] + dp2[k + 1][j] + sum[j] - sum[i - 1]);
}
}
}
对于本题的环形,可以考虑破环为链,链的长度为2n,比如1,2,3,4形成的环,改为1,2,3,4,1,2,3,4的链,这样任意取长度为4的区间,都可以看作是原来的环(相当于对起始点的枚举),最后对每一个dp[i][i + n - 1]枚举即可得到答案。
代码
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 201;
int a[N], sum[N],dp1[N][N], dp2[N][N];
int n, minn = 0x3f3f3f3f, maxx = 0;
int main() {
ios::sync_with_stdio(false);
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) {
sum[i] = a[i] + sum[i - 1];
}
memset(dp1, 0x3f, sizeof(dp1));
for (int i = 1; i <= 2 * n; ++i) {
dp1[i][i] = 0;
}
for (int len = 2; len <= n; ++len) {
for (int i = 1; i <= 2 * n - len + 1; ++i) {
int j = i + len - 1;
for (int k = i; k < j; ++k) {
dp1[i][j] = min(dp1[i][j], dp1[i][k] + dp1[k + 1][j] + sum[j] - sum[i - 1]);
dp2[i][j] = max(dp2[i][j], dp2[i][k] + dp2[k + 1][j] + sum[j] - sum[i - 1]);
}
}
}
for (int i = 1; i < n; ++i) {
minn = min(minn, dp1[i][i + n - 1]);
maxx = max(maxx, dp2[i][i + n - 1]);
}
cout << minn << endl;
cout << maxx << endl;
return 0;
}