对于这个问题,显然可以进行DP:
令dp[i]表示到i结尾的字符串可以表示的不同含义数,那么考虑两种转移:
末尾不替换含义:dp[i - 1]
末尾替换含义:dp[i - |B|] (A.substr(i - |B| + 1,|B|) = B)
那么对于末尾替换含义的转移,需要快速判断B能不能和当前位置的后缀匹配,kmp或者hash判断即可。
复杂度:O(N)
题解是说要kmp,但按它的做法实现起来kmp又不是必须的。
首先 dp[ i ] = dp[ i - 1] ;不管 i 位置匹配不匹配,它都应该继承前一位置的总数。
然后对于第 i 个位置,如果它从 i - m 到 第 i -1个位置正好和 B 串匹配,则 dp[ i ] += dp[ i -m ] ;
因为从i-m到i-1长度为m的这个串,如果它使用了第二个含意,则i-m到i-1这些位置都不能再变了,所以是加上第i - m个位置上的总数;
对于第i个位置是否“匹配”,可以用kmp预处理出来,也可以临时直接判断;
【代码】(附上优秀KMP模版)
/* ***********************************************
Author :angon
2016年7月29日 01:10:05
************************************************ */
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define REP(i,k,n) for(int i=k;i<n;i++)
#define REPP(i,k,n) for(int i=k;i<=n;i++)
#define scan(d) scanf("%d",&d)
#define scann(n,m) scanf("%d%d",&n,&m)
#define mst(a,k) memset(a,k,sizeof(a));
#define LL long long
#define N 100005
#define mod 1000000007
/*
inline int read()
{
int s=0;
char ch=getchar();
for(; ch<'0'||ch>'9'; ch=getchar());
for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';
return s;
}
inline void print(int x)
{
if(!x)return;
print(x/10);
putchar(x%10+'0');
}
*/
string a,b;
LL dp[100005];
int Next[N],idx[N];
void MakeNext(int m)
{
Next[0]=-1;
int i=0,j=-1;
while(i<m)
{
if(j==-1||b[i]==b[j])
{
++i,++j;
Next[i]=j;
}
else j=Next[j];
}
}
int KMP(int n,int m) //a是主串,b是模式串,n是主串长,m是模式串长
{
MakeNext(m);
int i=0,j=0,ret=0;
while(i<n)
{
if(a[i]==b[j]||j==-1)++i,++j;
else j=Next[j];
if(j==m)
{
idx[ret++] = i-m; //各匹配到的模式串在主串中的首字母位置
j=Next[j]; //允许在主串中重叠查找
/*j=0; //不允许重叠*/
}
}
return ret; //成功匹配次数
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int t;
scan(t);
for(int cas=1;cas<=t;cas++)
{
cin>>a>>b;
int len = a.length();
int len2 = b.length();
dp[0]=1;
for(int i=1;i<=len;i++)
{
dp[i] = dp[i-1];
if(i>=len2 && a.substr(i-len2,len2)==b)
dp[i] = ( dp[i] + dp[i-len2] ) % mod;
}
printf("Case #%d: %I64d\n",cas,dp[len]);
}
return 0;
}