本题为原洛谷原题,代码大部分是一位网友博客的改编版,多加一个t组循环和初始化,之前确实没见过这种的背包。
有点状态机配背包的味道。(刚学完背包和状态机的我表示我没学明白,当时只用分组背包求最优解。。。)
从新理思路的话终点看看代码中最后三层for循环中注释内容(原代码),和未注释内容(根据思路改变一下顺序)都对,其实因为dpn中一直存的值都要大于等于dpy,所以最后一个更新用fy更新fn似乎没神魔用。
#include <bits/stdc++.h>
using namespace std;
int n, m, k;
int dpy[205][205], dpn[205][205], fy[205][205], fn[205][205];
int a[205][205], b[205][205];
int main() {
int t;scanf("%d",&t);
while(t--)
{
memset(dpy,0,sizeof(dpy));
memset(dpn,0,sizeof(dpn));
memset(fy,0,sizeof(fy));
memset(fn,0,sizeof(fn));
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
scanf("%d%d%d", &n, &m, &k);
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++) {
char opt;
scanf("%d %c", &a[i][j], &opt);
if (opt == 'N')
b[i][j] = 0;
else
b[i][j] = 1;
}
for (int j = 1; j <= m; j++) {
int cnt = 0;
for (int i = n; i >= 1; i--) {
if (b[i][j])//如果这个位置是Y打穿。
dpn[j][cnt] += a[i][j];
else { //此位置是N,cnt++;多消耗一颗子弹。
cnt++;
dpy[j][cnt] = dpn[j][cnt - 1] + a[i][j];
dpn[j][cnt] = dpn[j][cnt - 1] + a[i][j];
}
}
}
for (int i = 1; i <= m; i++)
for (int j = 0; j <= k; j++) //前i列共用的子弹
for (int q = 0; q <= n && q <= j; q++) { //当前用的子弹
// fn[i][j] = max(fn[i][j],
// fn[i - 1][j - q] +
// dpn[i][q]); //最后一颗子弹不在当前列和前一列
// if (q != 0)
// fy[i][j] = max(fy[i][j],
// fn[i - 1][j - q] + dpy[i][q]); // 后打当前列
// if (j - q > 0)
// fy[i][j] = max(fy[i][j],
// fy[i - 1][j - q] + dpn[i][q]); //先打当前列
if(q>0)
fy[i][j]=max(fy[i][j],fn[i-1][j-q]+dpy[i][q]);//后打当前列。
if(j-q>0)
fy[i][j]=max(fy[i][j],fy[i-1][j-q]+dpn[i][q]); //先打当前列。
fn[i][j]=max(fn[i][j],fn[i-1][j-q]+dpn[i][q]);//最优一颗子弹不在当前列并且也不在前一列。
fn[i][j]=max(fn[i][j],fy[i][j]);
}
printf("%d\n", fy[m][k]);//答案只能由fy中存有,因为f[m][k]中存在1到m列都不是最后一列被打的局面,这种局面必然要大于等于有最优一列被打的状态值。
}
return 0;
}