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

来源：互联网发布：淘宝网店招的图片尺寸编辑：程序博客网时间：2024/05/16 09:37

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

2 0