【区间DP】洛谷P1880石子合并题解
题目传送门
分析
典型的区间DP,我们可以建立状态f[i][j]表示将区间[i,j]内的石子合并的最大值。那么我们可以把[i,j]分为[i,k]和[k+1,j](i<=k<j)两个区间,分别合并后求出最大值。
那么可以推出状态转移方程:
其中s数组是表示前缀和的,提前预处理了区间内的石子合并总分。
另外有一个要注意的地方,如果直接从前往后枚举区间的左右端点,那么因为k>=i,则会出现f[k+1][j]还未进行转移的情况,会导致莫名其妙的错误,所以可以枚举右端点和区间长度,来计算左端点即可。
最后统计整个数组的最小值时,要注意所需要的区间是[i,i+n-1]。
代码
#include <bits/stdc++.h>
#define INF (int)1e9
using namespace std;
int a[210];
int s[210];
int f1[210][210], f2[210][210];
int main()
{
int n;
cin >> n;
for (int i = 1; i <= n; ++i) {
cin >> a[i];
a[i+n] = a[i];
s[i] = s[i-1]+a[i];
}
for (int i = n+1; i <= 2*n; ++i) {
s[i] = s[i-1] + a[i];
}
for(int p=1;p<n;p++)
{
for(int i=1,j=i+p;(j<n+n) && (i<n+n);i++,j++)
{
f2[i][j]=INF;
for(int k=i;k<j;k++) {
f1[i][j] = max(f1[i][j], f1[i][k] + f1[k + 1][j] + s[j] - s[i - 1]);
f2[i][j] = min(f2[i][j], f2[i][k] + f2[k + 1][j] + s[j] - s[i - 1]);
}
}
}
int m1 = 0,m2 = INF;
for (int i = 1; i <= n; ++i) {
m1 = max(m1, f1[i][i+n-1]);
m2 = min(m2, f2[i][i+n-1]);
}
cout << m2 << endl;
cout << m1 << endl;
return 0;
}