dp区间问题
题目大意:设有N堆沙子排成一排,其编号为1,2,3,…,N(N<=300)。每堆沙子有一定的数量,可以用一个整数来描述,现在要将这N堆沙子合并成为一堆,每次只能合并相邻的两堆,合并的代价为这两堆沙子的数量之和,合并后与这两堆沙子相邻的沙子将和新堆相邻,合并时由于选择的顺序不同,合并的总代价也不相同,如有4堆沙子分别为 1 3 5 2 我们可以先合并1、2堆,代价为4,得到4 5 2 又合并 1,2堆,代价为9,得到9 2 ,再合并得到11,总代价为4+9+11=24,如果第二步是先合并2,3堆,则代价为7,得到4 7,最后一次合并代价为11,总代价为4+7+11=22;问题是:找出一种合理的方法,使总的代价最小。输出最小代价。
区间DP通用的转移方程如下:
f(i,j) = min{f[i,k] + f[k+1,j] + cost(i,j)
其中cost为将区间i~j合并起来的代价,可以用前缀和来计算(前缀和传送门)。
区间DP中要比较小心的是阶段的划分,本题中使用区间长度len作为阶段,为什么要选用len呢?单纯的看状态转移方程,我们很难确定如何划分阶段,因为里面没有任何关于len的信息。
对于使用递推的方式来解决DP问题,我们都需要从初始值推导出后续值,比如从0一直到N,而不能反着来。区间DP也有同样的处理方式,我们必须先从区间长度为0的初始值出发,也就是说,f(i,j)其实可以等价于一维的F(len)。
F(0) => f(0,0) f(1,1) f(2,2) ...... f(n,n)
那么使用len作为阶段就很正常的。
实现过程中除了注意len作为阶段,还需要注意初始化f[k][k]=0, 而f[i][j]初始化为比较大的数字。
打印语句可以比较清晰的看出是如何递推的。
示例:
4
1 3 5 2
#include <string.h> #include <cmath> #include <cstdio> #include <iostream> using namespace std; #define N 1100 int a[1100]; int s[1100]; int f[N][N]; int main() { #ifdef __MSYS__ freopen("test.txt", "r", stdin); #endif int n; scanf("%d", &n); // 注意初始化为比较大的数字 memset(f, 0x3f, sizeof(f)); for (int i = 1; i <= n; ++i) { scanf("%d", &a[i]); s[i] = s[i - 1] + a[i]; // l和r相等不需要合并,为0 f[i][i] = 0; printf("s[%d]=%d a[%d]=%d\n", i, s[i], i, a[i]); } // 计算阶段,高阶依赖低阶,以长度为阶 for (int len = 2; len <= n; ++len) { // len=2 1/2/3 =(4-2+1) for (int l = 1; l <= n - len + 1; ++l) { // left int r = l + len - 1; // right // for (int k = l; k < r; ++k) { // 计算最小值 f[l][r] = std::min(f[l][r], f[l][k] + f[k + 1][r]); printf("f[%d][%d] -> f[%d][%d]+f[%d][%d]\n", l, r, l, k, k+1, r); } // 加上cost f[l][r] += s[r] - s[l - 1]; printf(" f[%d][%d]=%d\n", l, r, f[l][r]); } } printf("%d\n", f[1][n]); return 0; }