题目背景
NOIP2015 提高组 Day2 T2
题目描述
有两个仅包含小写英文字母的字符串 A 和 B 。现在要从字符串 A 中取出 k 个互不重叠的非空子串,然后把这 k 个子串按照其在字符串 A 中出现的顺序依次连接起来得到一个新的字符串,请问有多少种方案可以使得这个新串与字符串 B 相等?注意:子串取出的位置不同也认为是不同的方案。
输入格式
第一行是三个正整数 n,m,k,分别表示字符串 A 的长度,字符串 B 的长度,以及问题描述中所提到的 k ,每两个整数之间用一个空格隔开。
第二行包含一个长度为 n 的字符串,表示字符串 A 。
第三行包含一个长度为 m 的字符串,表示字符串 B 。
输出格式
输出共一行,包含一个整数,表示所求方案数。由于答案可能很大,所以这里要求输出答案对 1,000,000,007 取模的结果。
样例数据 1
输入
6 3 1
aabaab
aab
输出
2
样例数据 2
输入
6 3 2
aabaab
aab
输出
7
样例数据 3
输入
6 3 3
aabaab
aab
输出
7
备注
【样例说明】
所有合法方案如下:(加下划线的部分表示取出的子串)
样例1:aab aab / aab aab
样例2:a ab aab / a aba ab / a a ba ab / aab a ab
aa b aab / aa baa b / aab aa b
样例3:a a b aab / a a baa b / a ab a a b / a aba a b
a a b a a b / a a ba a b / aab a a b
【数据范围】
对于第1组数据:1≤n≤500,1≤m≤50,k=1;
对于第2组至第3组数据:1≤n≤500,1≤m≤50,k=2;
对于第4组至第5组数据:1≤n≤500,1≤m≤50,k=m;
对于第1组至第7组数据:1≤n≤500,1≤m≤50,1≤k≤m;
对于第1组至第9组数据:1≤n≤1000,1≤m≤100,1≤k≤m;
对于所有10组数据:1≤n≤1000,1≤m≤200,1≤k≤m。
解析:
令表示到A串的第个位置,匹配到B串的第个位置,已经用个字串时,进行与不进行匹配的方案数。
但是这样做空间开不下,观察到每次转移都是从的状态转移过来的,所以考虑用滚动数组降维。
代码:
#include <bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int Max=1010;
int n,m,k,now;
long long f[2][205][205][2];
char a[Max],b[205];
int main()
{
scanf("%d%d%d\n%s\n%s",&n,&m,&k,a+1,b+1);
int now=1,pre=0;
f[0][0][0][0]=f[1][0][0][0]=1;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
for(int g=1;g<=k;g++)
{
if(a[i]==b[j]) f[now][j][g][1]=(f[pre][j-1][g][1]+f[pre][j-1][g-1][1]+f[pre][j-1][g-1][0])%mod;
else f[now][j][g][1]=0;
f[now][j][g][0]=(f[pre][j][g][0]+f[pre][j][g][1])%mod;
}
swap(now,pre);
}
cout<<(f[pre][m][k][0]+f[pre][m][k][1])%mod;
return 0;
}