Bzoj3473:字符串:广义后缀自动机

题目链接：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);
	}
}