题目链接:https://www.oj.swust.edu.cn:50443/problem/show/2806
大意:把目标串分隔成若干个子串,使这些子串都能与某个keyword对应,求分隔的方案数。
思路:如下图,当处理到结点i时,假设存在2~i,4~i,8~i三个前缀(且为keyword),我们可以分别从2、4、8分隔,则右边为1一个keyword,1乘以左边部分分隔的方案数,最后相加就是0~i分隔的方案数了。所以按照dp的思想从左到右处理一遍就可以了
然后可以用last数组优化找有效前缀的部分,即借助last数组直接O(1)找到为keyword的最长后缀。//加last数组后,在原本kmp中改动的两部分:61~64、76~80
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <queue>
using namespace std;
typedef long long LL;
const int manx=1e6+10;
const int max_len=1e5+10;
const int mod=1e9+7;
int cou,n,m;
int tree[manx][26],las[manx];
LL vs[manx],dp[max_len],fail[manx];
char str[max_len];
int newnode()
{
for(int i=0; i<26; i++)
tree[cou][i]=0;
las[cou]=fail[cou]=vs[cou++]=0;
return cou-1;
}
void init()
{
cou=0;
newnode();
}
void iinsert(char* s)
{
int u=0;
int len=strlen(s);
for(int i=0; i<len; i++)
{
int net=s[i]-'a';
if(!tree[u][net])
tree[u][net]=newnode();
u=tree[u][net];
}
vs[u]=len;
}
void getfail()
{
int u=0;
queue<int>qu;
for(int i=0; i<26; i++)
if(tree[0][i])
qu.push(tree[0][i]);
while(!qu.empty())
{
u=qu.front();
qu.pop();
for(int i=0; i<26; i++)
{
if(tree[u][i])
{
fail[tree[u][i]]=tree[fail[u]][i];
qu.push(tree[u][i]);
}
else
tree[u][i]=tree[fail[u]][i];
if(fail[tree[u][i]]&&vs[fail[tree[u][i]]])
las[tree[u][i]]=fail[tree[u][i]];
else
las[tree[u][i]]=las[fail[tree[u][i]]];
}
}
}
void query()
{
int u=0;
for(LL i=1; i<=n; i++)
{
dp[i]=0;
int net=str[i]-'a';
u=tree[u][net];
for(int j=u; j; j=las[j])
{
dp[i]+=dp[i-vs[j]];
dp[i]%=mod;
}
printf("%lld%c",dp[i],i==n?'\n':' ');
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
init();
dp[0]=1;
scanf("%d%d",&n,&m);
scanf("%s",str+1);
char s[max_len];
for(int i=0; i<m; i++)
{
scanf("%s",s);
iinsert(s);
}
getfail();
query();
}
return 0;
}