和Leo一起做爱字符串/线段树的好孩子CF700E Cool Slogans

这道题愈发加深了我对right集合的理解

也就是endpos集合

endpos集合是指一些可以以这个点结尾的合法集合

而维护使用经典的主席树

按照Parent树从上向下进行树形DP

使用主席树按top序合并pos集合

由于树上路径一定是当前的一个后缀所以必然出现了至少一次

这个时候查询posA-lenA+lenB,posB-1是否存在一个节点

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int N=5e5;
char S[N]={};
int n;
struct SAM{
    struct Node{
        int step,pre,pos;
        int vis[26];
    }SA[N];
    int last,cnt;
    void Insert(char C,int Id){
        int np=++cnt;
        int p=last;
        last=np;
        SA[np].step=SA[p].step+1;
        SA[np].pos=Id;
        for(;p&&!SA[p].vis[C-'a'];p=SA[p].pre)SA[p].vis[C-'a']=np;
        if(!p)SA[np].pre=1;
        else{
            int q=SA[p].vis[C-'a'];
            if(SA[q].step==SA[p].step+1){
                SA[np].pre=q;
            }
            else{
                int nq=++cnt;
                SA[nq].step=SA[p].step+1;
                SA[nq].pos=SA[q].pos;
                memcpy(SA[nq].vis,SA[q].vis,sizeof(SA[q].vis));
                SA[nq].pre=SA[q].pre;
                SA[q].pre=SA[np].pre=nq;
                for(;p&&SA[p].vis[C-'a']==q;p=SA[p].pre)SA[p].vis[C-'a']=nq;
            }
        }
    }
}T;
int lson[N*20];
int rson[N*20];
int root[N*20];
int cnt=0;
void Update(int &x,int l,int r,int pos){
    if(!x)x=++cnt;
    if(l==r)return;
    int mid=(l+r)/2;
    if(pos<=mid)Update(lson[x],l,mid,pos);
    else Update(rson[x],mid+1,r,pos);
}
int Merge(int x,int y){
    if(!x||!y)return x|y;
    int C=++cnt;
    lson[C]=Merge(lson[x],lson[y]);
    rson[C]=Merge(rson[x],rson[y]);
    return C;
}
int Query(int x,int l,int r,int L,int R){
    if(!x)return 0;
    if(L<=l&&r<=R)return 1;
    int mid=(l+r)/2;
    int ret=0;
    if(L<=mid)ret+=Query(lson[x],l,mid,L,R);
    if(mid< R)ret+=Query(rson[x],mid+1,r,L,R);
    return ret;
}
int C[N]={};
int A[N]={};
int top[N]={};
void Get_Top(){
    memset(C,0,sizeof(C));
    for(int i=1;i<=T.cnt;++i)C[T.SA[i].step]++;
    for(int i=1;i<=T.cnt;++i)C[i]+=C[i-1];
    for(int i=1;i<=T.cnt;++i)A[C[T.SA[i].step]--]=i;
}
int F[N]={};
int main(){
//	freopen("test.in","r",stdin);
//	freopen("test.out","w",stdout);
    scanf("%d",&n);
    scanf("%s",S+1);
    T.cnt=T.last=1;
    for(int i=1;i<=n;++i)T.Insert(S[i],i),Update(root[T.last],1,n,i);
    Get_Top();
    for(int i=T.cnt;i>1;i--){
        root[T.SA[A[i]].pre]=Merge(root[T.SA[A[i]].pre],root[A[i]]);	
    }
    int ans=1;
    for(int i=2;i<=T.cnt;++i){
        int u=A[i];
        int fa=T.SA[u].pre;
        if(fa==1){
            top[u]=u;
            F[u]=1;
            continue;
        }
        if(Query(root[top[fa]],1,n,T.SA[u].pos-T.SA[u].step+T.SA[top[fa]].step,T.SA[u].pos-1))F[u]=F[fa]+1,top[u]=u;
        else top[u]=top[fa],F[u]=F[fa];
        ans=max(ans,F[u]);
    }
    cout<<ans;
}