题意:
一句话可能有两个意思,给出两个字符串A,B。B是双重意思的字符串,问A这个字符串可有几种意思?也就是说,把B串在A中替换成*,一共可以构造出几种?
例:
hehehehe
hehe
答案是5
分析:
这题当时是我做的,第一次next数组编译错误QAQ,第二次A了。
思路是:先KMP求出B串在A串中出现的位置,然后就想着用dp了。由于在A串中B串可能会重叠,所以先O(n)预处理一下当前位置前边第一个不会重叠的串的位置。然后dp,有两个决策:
f[i][0]表示当前位置不替换B
f[i][1]表示当前位置替换B
状态转移方程:
f[i][0]=(f[i-1][0]+f[i-1][1])%mod;
if(p[i]!=-1)f[i][1]=(f[p[i]][0]+f[p[i]][1])%mod;
else f[i][1]=1;
#include<bits/stdc++.h>
using namespace std;
const int mod=1000000007;
typedef long long ll;
const int N=1e5+5;
char s[N],t[N];
int slen,tlen;
int nex[N],p[N],f[N][2];
int a[N],cnt;
void getNext()
{
int j,k;
j=0;k=nex[0]=-1;
while(j<tlen){
if(k==-1||t[j]==t[k])nex[++j]=++k;
else k=nex[k];
}
}
void KMP_Index()
{
getNext();
int i,j=0;
cnt=0;
for( i=0;i<slen;i++){
while(j&&s[i]!=t[j])j=nex[j];
if(s[i]==t[j])j++;
if(j==tlen){
a[++cnt]=i-tlen+1;
j=nex[j];
}
}
}
int main()
{
int T;
scanf("%d",&T);
for(int cas=1;cas<=T;cas++){
printf("Case #%d: ",cas);
scanf("%s",s);
scanf("%s",t);
tlen=strlen(t);slen=strlen(s);
KMP_Index();
int ans=0;
if(cnt==0){
printf("1\n");continue;
}
int j=1;
p[1]=-1;
for(int i=2;i<=cnt;i++){
if(a[i]-a[j]<tlen){
p[i]=-1;
}
else{
while(a[i]-a[j]>=tlen)j++;
j--;
p[i]=j;
}
}
f[1][0]=f[1][1]=1;
for(int i=2;i<=cnt;i++){
f[i][0]=(f[i-1][0]+f[i-1][1])%mod;
if(p[i]!=-1)f[i][1]=(f[p[i]][0]+f[p[i]][1])%mod;
else f[i][1]=1;
}
printf("%d\n",(f[cnt][0]+f[cnt][1])%mod);
}
return 0;
}