Shooting Bricks

Shooting Bricks

link[Problem - 7110 (hdu.edu.cn)]

题解

受先发现Y不能等价于是免费的，因为只有当你手上至少有一发子弹的时候才可以取Y，而当一个N上有Y的时候可以取N的时候把Y都带走，因此重要的就在于最后一发子弹打在哪，在次之前Y都等价于免费的，也就是要看最后枪打在哪一列的收益和其他列的收益的总和最大，发先是一个多阶段决策最优化问题(dp)。

先预处理出来每一列的信息:

vn[i, j] 表示第i列打了j枪且包含最后一发子弹的情况

vy[i, j] 表示第i列打了j枪且不包含最后一发的情况

状态表示

fn[i, j] 表示前i列用了j发子弹且包含最后一发子弹

fy[i, ] 表示前i列用了j发子弹且不包含最后一发子弹

状态计算

对于fy[i, j] 因为不含最后一发子弹，所以只需要枚举最后一列用了少发就可以

fy[i, j] = max(fy[i, j], fy[i - 1, j - k] + vy[i, k]);

对于fn[[i, j] 分类来看

1. 最后一发子弹在第j列之前，那么第j列的Y都是免费的

   fn[i, j] = max(fn[i, j], fn[i - 1, j - k] + vy[i, k])

2. 最后一发子弹在第j列，那么第j列之前的Y都是免费的

   fn[i, j] = max(fn[i, j], fy[i - 1, j - k] + vn[i, k])

Code

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <set>
#include <queue>
#include <vector>
#include <map>
#include <unordered_map>
#include <cmath> 
#include <stack>
#include <iomanip>
#include <deque> 
#include <sstream>
#define x first
#define y second
using namespace std;
typedef long double ld;
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
typedef unsigned long long ULL;
const int N = 410, M = 2 * N, INF = 0x3f3f3f3f, mod = 1e9 + 7;
const double eps = 1e-8;
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
int h[N], e[M], ne[M], w[M], idx;
void add(int a, int b, int v = 0) {
    e[idx] = b, w[idx] = v, ne[idx] = h[a], h[a] = idx ++;
}
int n, m, k;
bool g[N][N];
int a[N][N], vn[N][N], vy[N][N], fn[N][N], fy[N][N];
char op;
int main() {
    ios::sync_with_stdio(false), cin.tie(0);
    int T;
    cin >> T;
    while (T -- ) {
        cin >> n >> m >> k;
        for (int i = 1; i <= n; i ++ )
            for (int j = 1; j <= m; j ++ ) {
                cin >> a[i][j] >> op;
                g[i][j] = (op == 'Y');
            }
        for (int i = 0; i < N; i ++ )
            for (int j = 0; j < N; j ++ )
                fn[i][j] = fy[i][j] = vn[i][j] = vy[i][j] = 0;
        for (int j = 1; j <= m; j ++ ) {
            int cnt = 0; // 当前列用了几发子弹
            for (int i = n; i; i -- ) 
                if (g[i][j]) vy[j][cnt] += a[i][j];
                else {
                    cnt ++;
                    vn[j][cnt] = vy[j][cnt - 1] + a[i][j];
                    vy[j][cnt] = vn[j][cnt]; 
                }
        }
        for (int i = 1; i <= m; i ++ )
            for (int j = 0; j <= k; j ++ )
                for (int l = 0; l <= min(j, n); l ++ ) {
                    fy[i][j] = max(fy[i][j], fy[i - 1][j - l] + vy[i][l]);
                    if (l) fn[i][j] = max(fn[i][j], fy[i - 1][j - l] + vn[i][l]); // 最后一枪在第i列
                    if (j != l) fn[i][j] = max(fn[i][j], fn[i - 1][j - l] + vy[i][l]); // 最后一枪在第i列前面
                }
        cout << fn[m][k] << endl;
    }
return 0;
}