牛客 - 牛牛的01限定串

题目描述

牛牛的01限定串

解法：DP（C++）

我们先考虑这样一个问题：如果 $t$ 串全是 ？

如样例一

8 6 2 1 1
10110011
????????

这时如果我们看做纸条问题，那么也就是：求一条从$(0,0)$到$(cnt0,cn{t}_1)$的路径，只能向右（取0）或向下（取1）走，使得路径上的点的权值和最大。

于是，我们就得一个 $(6+1)\times (2+1)$的棋盘，左上角记为$(0,0)$

那么，现在的问题是，从左上角走到右下角有多少种走法？好的，是$C(8,6)$种。

那用 6 个 0 和 2 个 1 组成不同的字符串，有多少种？巧了，也是$C(8,6)$种。

于是，纸条问题就和这道题对应起来了，这时很容易就想到用动态规划。

记$dp[x][y]$表示$(x,y)$的得分，$a[x][y]$表示到达$(x,y)$这一步的得分，那么就有：

$$dp[x][y]=max(dp[x-1][y],dp[x][y-1])+a[x][y]$$

对应题目，细化一下转移方程，$pos$表示当前在棋盘的位置对应到字符串的位置，$sum[x]$ 表示标准串 $s$，到第 $x$ 位置为止共有多少个 1， 满足 $s$ 与 $t$ 前缀 $x_1$ 相同等价于 $sum[x_1]==j$，后缀 $x_1+1～n$ 相同就等价于 $sum[n]-sum[x_1]==cn{t}_1-j$：

① $t[pos]==0$，即只能右走
$mmax[i][j]=max(mmax[i][j],mmax[i][j-1]+va{l}_{pre}\ast (sum[pos]==j)+va{l}_{suf}\ast (sum[n]-sum[pos]==cn{t}_1-j))$

② $t[pos]==1$，即只能右走
$mmax[i][j]=max(mmax[i][j],mmax[i-1][j]+va{l}_{pre}\ast (sum[pos]==j)+va{l}_{suf}\ast (sum[n]-sum[pos]==cn{t}_1-j))$

③ $t[pos]==?$，即可右可下
$mmax[i][j]=max(mmax[i-1][j],mmax[i][j-1])+va{l}_{pre}\ast (sum[pos]==j)+va{l}_{suf}\ast (sum[n]-sum[pos]==cn{t}_1-j)$

情况③可以由情况①②表出

最后就是编程上的一些细节：

• s = " "+s; t = " "+t; 我们在两个字符串前加入空格的原因和棋盘大小的生成原理是一样的，让下标从 1 开始才是我们处理的内容
• mmax[0][0] = mmin[0][0] = vs*(sum[n]==cnt1); 初始化时如果两字符串后缀 $1～n$ 是相同的，要计算上这一部分

#include <bits/stdc++.h>

using namespace std;

const int N = 1e3+10;
int n, cnt0, cnt1, vp, vs, pos, mmax[N][N], mmin[N][N], sum[N];
string s, t;

int main()
{
    cin >> n >> cnt0 >> cnt1 >> vp >> vs;
    cin >> s >> t;
    s = " "+s;
    t = " "+t;
    for(int i=1;i<=n;i++)
        sum[i] = sum[i-1]+(s[i]-'0');
    memset(mmax, -0x3f, sizeof(mmax));
    memset(mmin, 0x3f, sizeof(mmin));
    mmax[0][0] = mmin[0][0] = vs*(sum[n]==cnt1);
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=n;j++)
        {
            pos = i+j;
            if(t[pos]!='0'&&j)
            {
                 mmax[i][j] = max(mmax[i][j], mmax[i][j-1]+vp*(sum[pos]==j&&pos)+vs*(sum[n]-sum[pos]==cnt1-j&&pos!=n));
                 mmin[i][j] = min(mmin[i][j], mmin[i][j-1]+vp*(sum[pos]==j&&pos)+vs*(sum[n]-sum[pos]==cnt1-j&&pos!=n));
            }
            if(t[pos]!='1'&&i)
            {
                mmax[i][j] = max(mmax[i][j], mmax[i-1][j]+vp*(sum[pos]==j&&pos)+vs*(sum[n]-sum[pos]==cnt1-j&&pos!=n));
                mmin[i][j] = min(mmin[i][j], mmin[i-1][j]+vp*(sum[pos]==j&&pos)+vs*(sum[n]-sum[pos]==cnt1-j&&pos!=n));
            }
        }
    }
    cout << mmin[cnt0][cnt1] << " " << mmax[cnt0][cnt1] << endl;

    return 0;
}