题目链接:3473:字符串
先建立一个广义后缀自动机,什么是广义后缀自动机?就是所有主串一起建立的一个后缀自动机。
广义后缀自动机的建立很简单,对于每个串,该怎么增量建立自动机就怎么建立,只不过为每个节点维护一个set保存这个节点的状态在那些字符串中出现过。当一个串增量构建完毕后,将后缀自动机的last指针指向后缀自动机的根即可进行下一发字符串的增量构建,这样就建出来了一发广义后缀自动机。
当然也可以把所有字符串连起来,中间插入一个在所有字符串中没有出现过的字符,然后一口气建完也可以,但是这两种建法在统计答案上有所不同,这里我用第一种方法(第二种你也不会啊喂!)
考虑一个节点,如果他在x个字符串中出现过,那么他的fa指针所指向的节点所代表的状态出现过的次数一定不小于他
并且我们已经为每个节点维护了一个set来记录在那些字符串中出现过,那么我们只需要自下向上合并set集合即可,在这之前需要整理出parent树的具体形态,然后一遍dfs,逆序处理set的启发式合并即可
统计答案只需把每个字符串在自动机上跑,跑到一个节点发现出现次数<K就往fa指针那里跳,知道符合条件,这时候贡献的答案就是当前节点的l[]属性的值了
#include<set>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;
const int maxn=200010;
set<int>d[maxn];
set<int>::iterator it;
int n,K,tot=1,h[maxn],sum[maxn];
struct edge{int to,next;}G[maxn*6];
string s[maxn];
struct Suffix_Automation{
int son[maxn][26],l[maxn],tot;
int fa[maxn],last,root;
void init(){
tot=0; last=root=++tot;
}
void add(int w,int id){
int p=last,np=last=++tot; last=np; l[np]=l[p]+1; d[np].insert(id);
while (p&&!son[p][w]) son[p][w]=np,p=fa[p];
if (!p) fa[np]=root;
else{
int q=son[p][w];
if (l[q]==l[p]+1) fa[np]=q;
else{
int nq=++tot; l[nq]=l[p]+1; fa[nq]=fa[q];
for (int i=0;i<26;++i) son[nq][i]=son[q][i];
fa[q]=fa[np]=nq;
while (son[p][w]==q) son[p][w]=nq,p=fa[p];
}
}
}
}sam;
void add(int x,int y){
G[++tot].to=y;G[tot].next=h[x];h[x]=tot;
}
void dfs(int x){
for (int i=h[x];i;i=G[i].next){
int v=G[i].to; dfs(v);
if (d[x].size()<d[v].size())
swap(d[x],d[v]);
for (it=d[v].begin();it!=d[v].end();++it)
d[x].insert(*it);
} sum[x]=d[x].size();
}
int main(){
int __size__ = 20 << 20;
char *__p__ = (char*)malloc (__size__) + __size__;
__asm__("movl %0, %%esp" :: "r"(__p__));
scanf("%d%d",&n,&K);
sam.init();
for (int i=1;i<=n;++i){
cin>>s[i];
int len=s[i].length();
for (int j=0;j<len;++j) sam.add(s[i][j]-'a',i);
sam.last=1;
}
for (int i=1;i<=sam.tot;++i)
if (sam.fa[i]) add(sam.fa[i],i);
dfs(sam.root);
if (K>n){
for (int i=1;i<=n;++i) printf("0 ");
return 0;
}
for (int i=1;i<=n;++i){
LL ans=0;
int now=sam.root,len=s[i].length();
for (int j=0;j<len;++j){
now=sam.son[now][s[i][j]-'a'];
while (sum[now]<K) now=sam.fa[now];
ans+=sam.l[now];
}
printf("%lld ",ans);
}
}