LOJ#2059. 「TJOI / HEOI2016」字符串 後綴數組+二分+主席樹

後綴自動機的做法很顯然:建出後綴樹,線段樹合併,然後倍增的時候更新答案就行.

後綴數組的做法也挺顯然:二分答案,然後用主席樹判定一下 $[a,b-mid+1]$ 是否有值即可.    

code: 

#include <bits/stdc++.h> 
#define N 200006   
#define lson s[x].ls 
#define rson s[x].rs 
#define inf 1000000000 
#define setI(s) freopen(s".in","r",stdin)   
#define setO(s) freopen(s".out","w",stdout)  
using namespace std;  
int n,tot,m;   
int rt[N];   
int arr[N],rk[N],tp[N],tax[N],sa[N],height[N],Log[N],dp[20][N];    
void qsort() 
{
    for(int i=0;i<=m;++i)  tax[i]=0;   
    for(int i=1;i<=n;++i)  ++tax[rk[i]];   
    for(int i=1;i<=n;++i)  tax[i]+=tax[i-1];   
    for(int i=n;i>=1;--i)  sa[tax[rk[tp[i]]]--]=tp[i];    
}
void getsuf() 
{
    for(int i=1;i<=n;++i) rk[i]=arr[i],tp[i]=i; 
    qsort();   
    for(int k=1;k<=n;k<<=1) 
    {
        int p=0;   
        for(int i=n-k+1;i<=n;++i) tp[++p]=i;   
        for(int i=1;i<=n;++i) if(sa[i]>k) tp[++p]=sa[i]-k;    
        qsort(),swap(rk,tp),rk[sa[1]]=p=1;   
        for(int i=2;i<=n;++i)  
            rk[sa[i]]=(tp[sa[i-1]]==tp[sa[i]]&&tp[sa[i-1]+k]==tp[sa[i]+k])?p:++p;      
        if(p==n) break;    
        m=p;
    }  
    for(int i=1;i<=n;++i) rk[sa[i]]=i;   
    for(int i=1,k=0;i<=n;++i)
    { 
        if(k) --k;   
        int j=sa[rk[i]-1];   
        while(arr[i+k]==arr[j+k]) ++k;   
        height[rk[i]]=k;            
    }     
}               
void RMQ() 
{
    for(int i=2;i<=n;++i) Log[i]=Log[i>>1]+1;   
    for(int i=1;i<=n;++i) dp[0][i]=height[i];         
    for(int j=1;(1<<j)<=n;++j)  
        for(int i=1;i+(1<<j)-1<=n;++i)   
            dp[j][i]=min(dp[j-1][i],dp[j-1][i+(1<<j-1)]);          
}
int getmin(int l,int r) 
{
    int k=Log[r-l+1];    
    return min(dp[k][l],dp[k][r-(1<<k)+1]);   
}
struct data 
{
    int ls,rs,sum;   
    data() { ls=rs=sum=0; }   
}s[N*25];            
void pushup(int x) 
{
    s[x].sum=s[lson].sum+s[rson].sum;     
}
int update(int x,int l,int r,int p) 
{ 
    int now=++tot;    
    s[now]=s[x];         
    s[now].sum++;  
    if(l==r)  return now; 
    int mid=(l+r)>>1;   
    if(p<=mid)  s[now].ls=update(s[x].ls,l,mid,p);  
    else s[now].rs=update(s[x].rs,mid+1,r,p);      
    return now;   
}         
int query(int x,int y,int l,int r,int L,int R) 
{
    if(!x) return 0;  
    if(l>=L&&r<=R) return s[y].sum-s[x].sum;     
    int mid=(l+r)>>1,re=0;  
    if(L<=mid)  re+=query(s[x].ls,s[y].ls,l,mid,L,R);  
    if(R>mid)   re+=query(s[x].rs,s[y].rs,mid+1,r,L,R);   
    return re;  
}
void build(int &x,int l,int r) 
{
    s[x=++tot].sum=0;  
    if(l==r) return;   
    int mid=(l+r)>>1;  
    build(lson,l,mid),build(rson,mid+1,r);  
}
char str[N];  
int main()    
{  
    int Q; 
    scanf("%d%d%s",&n,&Q,str+1);    
    for(int i=1;i<=n;++i) arr[i]=str[i]-'a'+1;   
    m=27,getsuf(),RMQ(),build(rt[0],1,n); 
    for(int i=1;i<=n;++i)
        rt[i]=update(rt[i-1],1,n,sa[i]);      
    int a,b,c,d,l,r,mid,an,lp,rp,L,R; 
    while(Q--) 
    { 
        scanf("%d%d%d%d",&a,&b,&c,&d);              
        l=1,r=min(d-c+1,b-a+1),mid=0,an=0; 
        while(l<=r) 
        {
            mid=(l+r)>>1;  
            L=R=rk[c];   
            lp=1,rp=rk[c]-1;      
            while(lp<=rp) 
            {
                int tmp=(lp+rp)>>1;   
                if(getmin(tmp+1,rk[c])>=mid) L=tmp,rp=tmp-1;  
                else lp=tmp+1;       
            }  
            lp=rk[c]+1,rp=n;   
            while(lp<=rp) 
            {
                int tmp=(lp+rp)>>1;                                                               
                if(getmin(rk[c]+1,tmp)>=mid) R=tmp,lp=tmp+1;  
                
                else rp=tmp-1;  
            }  
            // [lp,rp]        
            if(query(rt[L-1],rt[R],1,n,a,b-mid+1))   
                an=mid,l=mid+1;   
            else r=mid-1; 
        }   
        printf("%d\n",an);   
    }
    return 0; 
}

  

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章