CodeForces - 666E: Forensic Examination (广义SAM 线段树合并)

题意：给定字符串S，然后M个字符串T。Q次询问，每次给出(L,R,l,r)，问S[l,r]在L到R这些T字符串中，在哪个串出现最多，以及次数。

思路：把所有串建立SAM，然后可以通过倍增走到[l,r]在SAM上的位置p，然后在这个位置p上求，求的过程就是一个线段树求区间最值。 现在的关键是得到线段树，这个线段树记录了endpos，这个可以用启发式合并。

注意现在是广义后缀自动机，不能用拓扑排序合并线段树。 必须用fail树DFS来合并。 不然会出错。 具体我也不知道，不过我估计是因为广义自动机里面有空串导致的。 因为空串导致x和fax的长度相同，这样拓扑就会出错。 而普通的SAM没有这种情况。

    #include<bits/stdc++.h>
    #define rep2(i,a,b) for(int i=a;i>=b;i--)
    #define rep(i,a,b) for(int i=a;i<=b;i++)
    using namespace std;
    const int maxn=;
    char c[maxn],t[maxn];
    int rt[maxn],pos[maxn],fcy[maxn],N,M,Q,tot;
    int f[maxn][],ans,num;
    struct in{
        int L,R,Mx,id;
    }s[maxn*];
    void update(int now)
    {
        if(s[s[now].L].Mx>=s[s[now].R].Mx) {
            s[now].Mx=s[s[now].L].Mx;
            s[now].id=s[s[now].L].id;
        }
        else {
            s[now].Mx=s[s[now].R].Mx;
            s[now].id=s[s[now].R].id;
        }
    }
    void ins(int &now,int L,int R,int p)
    {
        if(!now) now=++tot;
        if(L==R){  s[now].Mx++; s[now].id=L; return ;}
        int Mid=(L+R)>>;
        if(p<=Mid) ins(s[now].L,L,Mid,p);
        else ins(s[now].R,Mid+,R,p);
        update(now);
    }
    void merge(int &now,int x,int y,int L,int R)
    {
        if(!x||!y) { now= x|y; return ;}
        now=++tot;int Mid=(L+R)>>;
        if(L==R) { s[now].Mx=s[x].Mx+s[y].Mx; s[now].id=L; return ;}
        merge(s[now].L,s[x].L,s[y].L,L,Mid);
        merge(s[now].R,s[x].R,s[y].R,Mid+,R);
        update(now);
    }
    void query(int Now,int L,int R,int l,int r)
    {
        if(!Now||l>r) return ;
        if(s[Now].Mx<num) return ;
        if(l<=L&&r>=R) {
            if(s[Now].Mx>num) num=s[Now].Mx,ans=s[Now].id;
            else if(s[Now].Mx==num&&num!=) ans=min(ans,s[Now].id);
            return ;
        }
        int Mid=(L+R)>>;
        if(l<=Mid) query(s[Now].L,L,Mid,l,r);
        if(r>Mid) query(s[Now].R,Mid+,R,l,r);
    }
    struct SAM{
        int ch[maxn][],fa[maxn],maxlen[maxn],cnt,last;
        int a[maxn],b[maxn];
        void init()
        {
            cnt=last=;
            memset(ch[],,sizeof(ch[]));
        }
        void add(int x,int id)
        {
            int np=++cnt,p=last; last=np;
            if(id>) ins(rt[np],,M,id);
            maxlen[np]=maxlen[p]+; memset(ch[np],,sizeof(ch[np]));
            while(p&&!ch[p][x]) ch[p][x]=np,p=fa[p];
            if(!p) fa[np]=;
            else {
                int q=ch[p][x];
                if(maxlen[q]==maxlen[p]+) fa[np]=q;
                else {
                   int nq=++cnt; maxlen[nq]=maxlen[p]+;
                   fa[nq]=fa[q]; fa[q]=fa[np]=nq;
                   memcpy(ch[nq],ch[q],sizeof(ch[q]));
                    while(p&&ch[p][x]==q) ch[p][x]=nq,p=fa[p];
                }
            }
        }
        vector<int>G[maxn];
        void dfs(int u)
        {
            for(int i=;i<G[u].size();i++){
                int v=G[u][i];
                f[v][]=u; dfs(v);
                merge(rt[u],rt[u],rt[v],,M);
            }
        }
        void DFS()
        {
            rep(i,,cnt) G[fa[i]].push_back(i);
            dfs();
        }
    }T;
    void solve()
    {
        int L,R,l,r,len;
        scanf("%d%d%d%d",&L,&R,&l,&r);
        len=r-l+;
        int now=pos[r];
        if(T.maxlen[now]<len) {
            printf("%d %d\n",L,);
            return ;
        }
        for(int i=;i>=;i--)
            if(T.maxlen[f[now][i]]>=len) now=f[now][i];
        ans=L; num=;
        query(rt[now],,M,L,R);
        printf("%d %d\n",ans,num);
    }
    int main()
    {
        T.init();
        scanf("%s",t+);
        N=strlen(t+); T.last=;
        rep(i,,N) T.add(t[i]-'a',),pos[i]=T.last;
        scanf("%d",&M);
        rep(i,,M) {
            T.last=;
            scanf("%s",c+); N=strlen(c+);
            rep(j,,N) T.add(c[j]-'a',i);
        }
        T.DFS();
        rep(i,,) rep(j,,T.cnt) f[j][i]=f[f[j][i-]][i-];
        scanf("%d",&Q);
        while(Q--) solve();
        return ;
    }

稍微优化一下，为了省空间和时间，也可以手动插入S，即S没有必要插入SAM中。

#include<bits/stdc++.h>
#define rep2(i,a,b) for(int i=a;i>=b;i--)
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=;
char c[maxn],t[maxn];
int rt[maxn],pos[maxn],fcy[maxn],N,M,Q,tot;
int f[maxn][],ans,num;
struct in{
    int L,R,Mx,id;
}s[maxn*];
void update(int now)
{
    if(s[s[now].L].Mx>=s[s[now].R].Mx) {
        s[now].Mx=s[s[now].L].Mx;
        s[now].id=s[s[now].L].id;
    }
    else {
        s[now].Mx=s[s[now].R].Mx;
        s[now].id=s[s[now].R].id;
    }
}
void ins(int &now,int L,int R,int p)
{
    if(!now) now=++tot;
    if(L==R){  s[now].Mx++; s[now].id=L; return ;}
    int Mid=(L+R)>>;
    if(p<=Mid) ins(s[now].L,L,Mid,p);
    else ins(s[now].R,Mid+,R,p);
    update(now);
}
void merge(int &now,int x,int y,int L,int R)
{
    if(!x||!y) { now= x|y; return ;}
    now=++tot;int Mid=(L+R)>>;
    if(L==R) { s[now].Mx=s[x].Mx+s[y].Mx; s[now].id=L; return ;}
    merge(s[now].L,s[x].L,s[y].L,L,Mid);
    merge(s[now].R,s[x].R,s[y].R,Mid+,R);
    update(now);
}
void query(int Now,int L,int R,int l,int r)
{
    if(!Now||l>r) return ;
    if(s[Now].Mx<num) return ;
    if(l<=L&&r>=R) {
        if(s[Now].Mx>num) num=s[Now].Mx,ans=s[Now].id;
        else if(s[Now].Mx==num&&num!=) ans=min(ans,s[Now].id);
        return ;
    }
    int Mid=(L+R)>>;
    if(l<=Mid) query(s[Now].L,L,Mid,l,r);
    if(r>Mid) query(s[Now].R,Mid+,R,l,r);
}
struct SAM{
    int ch[maxn][],fa[maxn],maxlen[maxn],cnt,last;
    int a[maxn],b[maxn];
    void init()
    {
        cnt=last=;
        memset(ch[],,sizeof(ch[]));
    }
    void add(int x,int id)
    {
        int np=++cnt,p=last; last=np;
        if(id>) ins(rt[np],,M,id);
        maxlen[np]=maxlen[p]+; memset(ch[np],,sizeof(ch[np]));
        while(p&&!ch[p][x]) ch[p][x]=np,p=fa[p];
        if(!p) fa[np]=;
        else {
            int q=ch[p][x];
            if(maxlen[q]==maxlen[p]+) fa[np]=q;
            else {
               int nq=++cnt; maxlen[nq]=maxlen[p]+;
               fa[nq]=fa[q]; fa[q]=fa[np]=nq;
               memcpy(ch[nq],ch[q],sizeof(ch[q]));
                while(p&&ch[p][x]==q) ch[p][x]=nq,p=fa[p];
            }
        }
    }
    vector<int>G[maxn];
    void dfs(int u)
    {
        for(int i=;i<G[u].size();i++){
            int v=G[u][i];
            f[v][]=u; dfs(v);
            merge(rt[u],rt[u],rt[v],,M);
        }
    }
    void DFS()
    {
        rep(i,,cnt) G[fa[i]].push_back(i);
        dfs();
    }
}T;
void solve()
{
    int L,R,l,r,len;
    scanf("%d%d%d%d",&L,&R,&l,&r);
    len=r-l+;
    int now=pos[r];
    if(fcy[r]<len) {
        printf("%d %d\n",L,);
        return ;
    }
    for(int i=;i>=;i--)
        if(T.maxlen[f[now][i]]>=len) now=f[now][i];
    ans=L; num=;
    query(rt[now],,M,L,R);
    printf("%d %d\n",ans,num);
}
int main()
{
    T.init();
    scanf("%s",t+);
    scanf("%d",&M);
    rep(i,,M) {
        T.last=;
        scanf("%s",c+); N=strlen(c+);
        rep(j,,N) T.add(c[j]-'a',i);
    }
    int now=,len=; N=strlen(t+);
    rep(i,,N) {
        int x=t[i]-'a';
        if(T.ch[now][x]) now=T.ch[now][x],len++;
        else {
            while(now&&!T.ch[now][x]) now=T.fa[now];
            if(now==) now=,len=;
            else len=T.maxlen[now]+,now=T.ch[now][x];
        }
        pos[i]=now;
        fcy[i]=len;
    }
    T.DFS();
    rep(i,,) rep(j,,T.cnt) f[j][i]=f[f[j][i-]][i-];
    scanf("%d",&Q);
    while(Q--) solve();
    return ;
}