UOJ149【NOIP2015】子串 （DP）

UOJ149【NOIP2015】子串

原题地址：http://uoj.ac/problem/149

题意：
有两个仅包含小写英文字母的字符串 A 和 B。

现在要从字符串 A 中取出 k 个互不重叠的非空子串，然后把这 k 个子串按照其在字符串 A 中出现的顺序依次连接起来得到一个新的字符串。请问有多少种方案可以使得这个新串与字符串 B 相等？

注意：子串取出的位置不同也认为是不同的方案。

由于答案可能很大，所以这里要求输出答案对1000000007取模的结果。

数据范围
N<=1000，M<=200，K<=200

题解：
我写的相当朴素：
f[i][j][k][0/1]表示从A的1~i位取k个串，拼出B的1~j位，且A的第i位 取/不取的方案数。

初始值：f[0~n][0][0][0]=f[0][0~m][0][0]=1 其他为0。
for(i=1~N)
for(j=1~m)
for(k=1~k)
dp[i][j][k][0]= dp[i-1][j][k][1] + dp[i-1][j][k][0]
若A[i]==B[j]
dp[i][j][k][1]= dp[i-1][j-1][k][1] + dp[i-1][j-1][k-1][1] + dp[i-1][j-1][k-1][0]

数组第一维开不下要滚。
因为初始f[0][0~n][0][0]=f[0][0~m][0][0]=1 ，所以做完f[1][…][…][…][…]后记得清掉f[0][…][…][…][…]，然后把f[0][0][0][0]赋为1。

（K范围看小了，以为开的下，没滚，WA…）

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1005;
const int M=205;
const int K=205;
const int mod=1000000007;
char A[N],B[M];
int n,m,kk,dp[2][M][K][2];
int main()
{
    scanf("%d%d%d",&n,&m,&kk);
    scanf("%s",A+1); scanf("%s",B+1);
    memset(dp,0,sizeof(dp));
    int pre=0; int now=1;
    for(int i=0;i<=m;i++) dp[pre][i][0][0]=1;
    dp[now][0][0][0]=1;
    for(int i=1;i<=n;i++)
    {   
        for(int j=1;j<=min(m,i);j++)
        {
            for(int k=1;k<=kk;k++)
            {
                if(A[i]==B[j])
                {
                    dp[now][j][k][1]=(dp[pre][j-1][k-1][1]+dp[pre][j-1][k][1])%mod;
                    dp[now][j][k][1]=(dp[now][j][k][1]+dp[pre][j-1][k-1][0])%mod;
                }
                else dp[now][j][k][1]=0;
                dp[now][j][k][0]=(dp[pre][j][k][0]+dp[pre][j][k][1])%mod;
            }
        }
        if(i==1) 
        {
            memset(dp[pre],0,sizeof(dp[pre]));
            dp[pre][0][0][0]=1; 
        }
        swap(now,pre);
    }
    int ans=(dp[pre][m][kk][1]+dp[pre][m][kk][0])%mod;

    printf("%d\n",ans); 
    return 0;
}