挺好的一道题,就是要做一个last优化,每次的last要返回到之前的有值节点,也就是单词的尾的对应节点,然后就不会超时了…… 呜呜呜,之前一直超时,以为是初始化的memset()问题,以前被卡过memset(),然后发现其实要是有多个相同的节点,岂不是会反复无常的跑无用的fail指针,于是就用了last优化掉了,过了。过咯!
对了,用DP推状态,这个比划一下,挺简单的,就是推对应后缀的去掉后缀的字符串的状态的求和。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 1e5 + 7;
const int mod = 1e9 + 7;
int N, M;
char virus[maxN], sss[maxN];
ll dp[maxN];
struct node
{
node *next[26];
node *fail, *las; //las用来寻找上一个为sum==1的点用以优化路径
int sum, len;
node()
{
for(int i=0; i<26; i++) next[i] = NULL;
fail = NULL; las = NULL;
sum = len = 0;
}
};
node *root;
void update(char *s)
{
node *temp = root;
int len = (int)strlen(s);
for(int i=0; i<len; i++)
{
int x = s[i] - 'a';
if(temp->next[x] == NULL) temp->next[x] = new node();
temp = temp->next[x];
}
temp->sum = 1;
temp->len = len;
}
void build_fail()
{
queue<node *> Q;
Q.push(root);
node *temp, *p;
while(!Q.empty())
{
temp = Q.front(); Q.pop();
for(int i=0; i<26; i++)
{
if(temp->next[i])
{
if(temp == root)
{
temp->next[i]->fail = root;
temp->next[i]->las = root;
}
else
{
p = temp->fail;
while(p)
{
if(p->next[i])
{
temp->next[i]->fail = p->next[i];
break;
}
p = p->fail;
}
if(!p) temp->next[i]->fail = root;
else
{
if(p->next[i]->sum > 0) temp->next[i]->las = p->next[i];
else temp->next[i]->las = p->next[i]->las;
}
}
Q.push(temp->next[i]);
}
}
}
}
void AC_auto()
{
node *p = root;
for(int i=1; i<=N; i++)
{
int x = virus[i] - 'a';
while(p != root && !p->next[x]) p = p->fail;
p = p->next[x];
if(!p) p = root;
node *temp = p;
while(temp && temp != root)
{
if(temp->sum > 0)
{
dp[i] += dp[i - temp->len];
if(dp[i] >= mod) dp[i] -= mod;
}
temp = temp->las;
}
}
}
inline void init()
{
for(int i=1; i<=N; i++) dp[i] = 0;
dp[0] = 1;
root = (node *)malloc(sizeof(node));
for(int i=0; i<26; i++) root->next[i] = NULL;
root->fail = NULL; root->las = NULL;
root->sum = 0;
}
int main()
{
int T; scanf("%d", &T);
while(T--)
{
scanf("%d%d", &N, &M);
init();
scanf("%s", virus + 1);
for(int i=1; i<=M; i++)
{
scanf("%s", sss);
update(sss);
}
build_fail();
AC_auto();
for(int i=1; i<=N; i++) printf("%lld%c", dp[i], i==N?'\n':' ');
}
return 0;
}