给定n个串m个询问,问每个串在n个串多少个串中出现了。
构建广义后缀自动机,(就是把所有字符串的后缀自动机合并起来)其实只需要add的时候注意一下就可以了。
然后对于每一个串,跑一边匹配,到达了now点。
如果now等于零,那么无法匹配,为0
否则就是询问子树中属于不同的后缀的节点又多少个。
如果n的数目比较小,可以考虑O(nl)去树形DP
我们只能,用vector打上标记。
然后dfs序搞出来,然后就是区间不同数的计数问题,然后直接《HH的项链》这道题目一套就可以了。
这一题套一题确实很有趣,代码丑到爆炸
#include <map>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std; #define F(i,j,k) for (int i=j;i<=k;++i)
#define D(i,j,k) for (int i=j;i>=k;--i)
#define maxn 500005 int in[maxn],out[maxn],tot,lst[maxn],ans[maxn];
int arr[maxn],pos[maxn],top,nxt[maxn]; struct Query{
int l,r,id;
void print()
{
printf("Query About %d to %d\n",l,r);
}
}brr[maxn]; struct Generalized_Suffix_Automaton{
int h[maxn],to[maxn],ne[maxn],en;
char s[maxn];
int last,cnt,n,q;
map<int,int>go[maxn];
int l[maxn],fa[maxn];
vector<int>v[maxn];
void init(){last=cnt=1;memset(h,-1,sizeof h);}
void addedge(int a,int b)
{to[en]=b;ne[en]=h[a];h[a]=en++;}
void add(int x,int id)
{
int p=last,q;
if (q=go[p][x])
{
if (l[p]+1==l[q]) last=q;
else
{
int nq=++cnt; l[nq]=l[p]+1;
go[nq]=go[q];
fa[nq]=fa[q];
fa[q]=nq;
for (;p&&go[p][x]==q;p=fa[p]) go[p][x]=nq;
last=nq;
}
}
else
{
int np=++cnt;
l[np]=l[p]+1;
for (;p&&!go[p][x];p=fa[p]) go[p][x]=np;
if (!p) fa[np]=1;
else
{
q=go[p][x];
if (l[q]==l[p]+1) fa[np]=q;
else
{
int nq=++cnt;l[nq]=l[p]+1;
go[nq]=go[q];
fa[nq]=fa[q];
fa[q]=fa[np]=nq;
for (;p&&go[p][x]==q;p=fa[p]) go[p][x]=nq;
}
}
last=np;
}
v[last].push_back(id);
}
void dfs(int k)
{
in[k]=++tot;
for (int i=0;i<v[k].size();++i) arr[++tot]=v[k][i];
for (int i=h[k];i>=0;i=ne[i]) dfs(to[i]);
out[k]=tot;
}
void solve()
{
init();
scanf("%d%d",&n,&q);
F(i,1,n)
{
last=1;
scanf("%s",s+1);
int len=strlen(s+1);
F(j,1,len) add(s[j],i);
}
F(i,1,cnt) addedge(fa[i],i);
dfs(1);
F(i,1,q)
{
scanf("%s",s+1);
int len=strlen(s+1);
// printf("%s\n",s+1);
// F(j,1,len) printf("-%c-",s[j]); printf("\n");
int now=1;
F(j,1,len){now=go[now][s[j]];}
// printf("now is %d\n",now);
if (now)
{
++top;
brr[top].l=in[now];
brr[top].r=out[now];
brr[top].id=i;
}
}
}
}sam; struct Bit_Tree{
int x[maxn];
void add(int pos,int del)
{for (;pos<=tot;pos+=pos&(-pos)) x[pos]+=del;}
int query(int pos)
{int ret=0;for (;pos;pos-=pos&(-pos))ret+=x[pos];return ret;}
}BT; bool cmp(Query a,Query b)
{return a.l==b.l?a.r<b.r:a.l<b.l;} int main()
{
sam.solve();
// F(i,1,tot) printf("%d ",arr[i]); printf("\n");
// F(i,1,top) brr[i].print();
sort(brr+1,brr+top+1,cmp);
F(i,1,tot)
{
if (!lst[arr[i]]) BT.add(i,1);
else nxt[lst[arr[i]]]=i;
lst[arr[i]]=i;
}
int Added=1;
F(i,1,top)
{
while (Added<brr[i].l)
{
if (nxt[Added]&&arr[nxt[Added]]) BT.add(nxt[Added],1);
Added++;
}
ans[brr[i].id]=BT.query(brr[i].r)-BT.query(brr[i].l-1);
}
F(i,1,sam.q) printf("%d\n",ans[i]);
}