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;
}