【bzoj 3277】串 & 【bzoj 3473】字符串（后缀自动机）

传送门biu~biu~
对于所有串建广义后缀自动机，沿着父亲指针不断向root走，把路径上的节点计数+1。为了避免出现重复，记录每个节点最后一次被哪个串访问，串在走的时候如果遇到了这个串已经访问过的节点就停止。
可以在parent树上dp出这个点到root路径上有多少子串出现了k次以上作为答案。最后把每个串在自动机上跑一遍，过程中每个访问到的节点答案的和即为每个串的答案。
复杂度不会证。

#include<bits/stdc++.h>
using namespace std;
string s[100005];
int n,k;
struct Node{
    vector<Node*>to;
    Node *ch[26],*fa;
    int val,sum,vis;
    long long ans;
    Node(int _=0):val(_){
        sum=vis=ans=0,fa=0x0;
        memset(ch,0x0,sizeof ch);
    }
}*root=new Node,*last;
inline void extend(int z){
    Node *p=last,*np=last=new Node(p->val+1);
    while(p && !p->ch[z])   p->ch[z]=np,p=p->fa;
    if(!p){
        np->fa=root;
        root->to.push_back(np);
        return;
    }
    Node *q=p->ch[z];
    if(q->val==p->val+1){
        np->fa=q;
        q->to.push_back(np);
        return;
    }
    Node *nq=new Node(p->val+1);
    memcpy(nq->ch,q->ch,sizeof nq->ch);
    nq->fa=q->fa;q->fa->to.push_back(nq);
    q->fa=np->fa=nq;nq->to.push_back(q),nq->to.push_back(np);
    while(p && p->ch[z]==q) p->ch[z]=nq,p=p->fa;
}
void dfs(Node *o){
    if(o!=root){
        o->ans=(o->sum>=k)*(o->val-o->fa->val);
        o->ans+=o->fa->ans;
    } 
    for(int i=0;i<o->to.size();++i) 
        if(o->to[i]->fa==o) dfs(o->to[i]);
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i){
        cin>>s[i],last=root;
        for(int j=0;s[i][j];++j)    extend(s[i][j]-'a');
    }
    for(int i=1;i<=n;++i){
        Node *o=root;
        for(int j=0;s[i][j];++j){
            o=o->ch[s[i][j]-'a'];
            Node *p=o;
            while(p && p->vis!=i)   p->vis=i,++p->sum,p=p->fa;
        }
    }
    dfs(root);
    for(int i=1;i<=n;++i){
        Node *o=root;long long ans(0);
        for(int j=0;s[i][j];++j)
            o=o->ch[s[i][j]-'a'],ans+=o->ans;
        printf("%lld ",ans);
    }
    return 0;
}