【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;
}