存个板子,内容以后补
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+60;
int n,tot=1;
char s[maxn];
struct mint
{
int son[26],flag,fail;
}trie[maxn];
void insert(char *s)
{
int p=1,len=strlen(s);
for(int i=0;i<len;++i)
{
int ch=s[i]-'a';
if(!trie[p].son[ch]) trie[p].son[ch]=++tot;
p=trie[p].son[ch];
}
trie[p].flag++;
}
void getFail()
{
queue<int> q;
for(int i=0;i<26;++i) trie[0].son[i]=1;
q.push(1);trie[1].fail=0;
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=0;i<26;++i)
{
if(trie[u].son[i])
{
trie[trie[u].son[i]].fail=trie[trie[u].fail].son[i];
q.push(trie[u].son[i]);
}
else trie[u].son[i]=trie[trie[u].fail].son[i];
}
}
}
int query(char *s)
{
int p=1,ans=0,len=strlen(s);
for(int i=0;i<len;++i)
{
int v=s[i]-'a';
int k=trie[p].son[v];
while(k>1 && trie[k].flag!=-1)
{
ans+=trie[k].flag;
trie[k].flag=-1;
k=trie[k].fail;
}
p=trie[p].son[v];
}
return ans;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;++i)
{
scanf("%s",s);
insert(s);
}
getFail();
scanf("%s",s);
printf("%d\n",query(s));
return 0;
}
拓扑优化建图
#include<bits/stdc++.h>
using namespace std;
const int maxn=2e6+60;
int n,vis[maxn],Map[maxn],tot=1;
int in[maxn];
char s[maxn];
struct mint
{
int son[26],flag,ans,fail;
}trie[maxn];
void insert(char *s,int num)
{
int len=strlen(s),p=1;
for(int i=0;i<len;++i)
{
int v=s[i]-'a';
if(!trie[p].son[v]) trie[p].son[v]=++tot;
p=trie[p].son[v];
}
if(!trie[p].flag) trie[p].flag=num;
Map[num]=trie[p].flag;
}
void getFail()
{
for(int i=0;i<26;++i) trie[0].son[i]=1;
queue<int> q;
q.push(1);
while(!q.empty())
{
int u=q.front();
q.pop();
int Fail=trie[u].fail;
for(int i=0;i<26;++i)
{
int v=trie[u].son[i];
if(v)
{
trie[v].fail=trie[Fail].son[i];
in[trie[v].fail]++;
q.push(v);
}
else trie[u].son[i]=trie[Fail].son[i];
}
}
}
void query(char *s)
{
int len=strlen(s),p=1;
for(int i=0;i<len;++i)
{
int v=s[i]-'a';
p=trie[p].son[v];
trie[p].ans++;
}
}
void topo()
{
queue<int> q;
for(int i=1;i<=tot;++i) if(in[i]==0) q.push(i);
while(!q.empty())
{
int u=q.front();
q.pop();
vis[trie[u].flag]=trie[u].ans;
int v=trie[u].fail;
trie[v].ans+=trie[u].ans;
in[v]--;
if(in[v]==0) q.push(v);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;++i)
{
scanf("%s",s);
insert(s,i);
}
getFail();
scanf("%s",s);
query(s);
topo();
for(int i=1;i<=n;++i) printf("%d\n",vis[Map[i]]);
return 0;
}