BZOJ3926

最新推荐文章于 2019-09-13 10:39:58 发布

weixin_30654583

最新推荐文章于 2019-09-13 10:39:58 发布

阅读量142

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Xiaojianxiang/p/6638054.html

本文介绍了一道关于广义后缀自动机的板子题解决方案，通过具体实例讲解了如何利用广义后缀自动机来求解特定字符串问题，并提供了完整的C++代码实现。

来自蒟蒻XXJ的做题记录

%%%陈老师的语文水平

广义后缀自动机的板子题

最多有\(20\)个叶子

随手画一画以每个叶子为根进行\(dfs\) 到达叶子的路径的后缀全都是不同的后缀

直接建出来看一下有多少不一样的子串就好了

#include<bits/stdc++.h>
#define mem(i,j) memset(i,j,sizeof(i))
#define mcy(i,j) memcpy(i,j,sizeof(i))
#define GO(i,here) for(int i=head[here];i!=-1;i=nex[i])
using namespace std;
const int MAXN=100010;
inline int in(){
    int a(0);char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') a=(a<<1)+(a<<3)+c-'0',c=getchar();
    return a;
}
int n,c,ans;
int head[MAXN],to[2*MAXN],nex[2*MAXN],top1;
int ru[MAXN],col[MAXN];
void add(int a,int b){
    nex[top1]=head[a];head[a]=top1;to[top1++]=b;
    nex[top1]=head[b];head[b]=top1;to[top1++]=a;
}
struct SAM{
    int trans[40*MAXN][11],fa[40*MAXN],lth[40*MAXN];
    long long s[40*MAXN];
    int cnt;
    SAM(){
        cnt=0;++cnt;
        mem(trans,0);mem(fa,0);mem(lth,0);mem(s,0);
    }
    int extend(int x,int ww){
        int np=++cnt,p=x;
        lth[np]=lth[p]+1;
        for(;p&&!trans[p][ww];p=fa[p]) trans[p][ww]=np;
        if(!p) fa[np]=1;
        else{
            int q=trans[p][ww];
            if(lth[q]==lth[p]+1) fa[np]=q;
            else{
                int nq=++cnt;
                lth[nq]=lth[p]+1;
                mcy(trans[nq],trans[q]);
                fa[nq]=fa[q];
                fa[q]=fa[np]=nq;
                for(;p&&trans[p][ww]==q;p=fa[p]) trans[p][ww]=nq;
            }
        }
        return np;
    }
    void solve(){
        for(int i=1;i<=cnt;i++)
            ans+=lth[i]-lth[fa[i]];
    }
}sam;
void dfs(int here,int f,int x){
    int t=sam.extend(x,col[here]);
    GO(i,here){
        if(to[i]==f) continue;
        dfs(to[i],here,t);
    }
}
void input(){
    mem(head,-1);
    n=in();c=in();
    for(int i=1;i<=n;i++) col[i]=in();
    for(int i=1;i<n;i++){
        int a,b;
        a=in();b=in();
        add(a,b);
        ++ru[a];++ru[b];
    }
    for(int i=1;i<=n;i++)
        if(ru[i]==1) dfs(i,0,1);
}
int rk[40*MAXN];
bool cmp(int i,int j){
    return sam.lth[i]>sam.lth[j];
}
void init(){
    for(int i=1;i<=sam.cnt;i++) rk[i]=i;
    sort(rk+1,rk+sam.cnt+1,cmp);
    for(int i=1;i<=sam.cnt;i++){
        int ind=rk[i];
        sam.s[ind]=1;
        for(int j=0;j<c;j++){
            if(sam.trans[ind][j]) sam.s[ind]+=sam.s[sam.trans[ind][j]];
        }
    }
}
void xxj(){
    //sam.solve();
    init();
}
void output(){
    cout<<sam.s[1]-1<<endl;
}

int main(){
    input();
    xxj();
    output();
    return 0;
}

转载于:https://www.cnblogs.com/Xiaojianxiang/p/6638054.html