Bzoj 4556: [Tjoi2016&Heoi2016]字符串

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100010
int root[MAXN],Ws[MAXN],wa[MAXN],wb[MAXN],sa[MAXN],wv[MAXN],Rank[MAXN],height[MAXN],MN[MAXN][17],sum[MAXN*18],ls[MAXN*18],rs[MAXN*18],SIZE;
char str[MAXN];
int read()
{
    int s=0,fh=1;char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)fh=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘){s=s*10+(ch-‘0‘);ch=getchar();}
    return s*fh;
}
int cmp(int *r,int a,int b,int l){return (r[a]==r[b])&&(r[a+l]==r[b+l]);}
void DA(char *r,int *sa,int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0;i<m;i++)Ws[i]=0;
    for(i=0;i<n;i++)Ws[x[i]=r[i]]++;
    for(i=0;i<m;i++)Ws[i]+=Ws[i-1];
    for(i=n-1;i>=0;i--)sa[--Ws[x[i]]]=i;
    for(j=1,p=1;p<n;j*=2,m=p)
    {
        for(p=0,i=n-j;i<n;i++)y[p++]=i;
        for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
        for(i=0;i<n;i++)wv[i]=x[y[i]];
        for(i=0;i<m;i++)Ws[i]=0;
        for(i=0;i<n;i++)Ws[wv[i]]++;
        for(i=0;i<m;i++)Ws[i]+=Ws[i-1];
        for(i=n-1;i>=0;i--)sa[--Ws[wv[i]]]=y[i];
        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}
void calheight(int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++)Rank[sa[i]]=i;
    for(i=0;i<n;height[Rank[i++]]=k)
        for(k?k--:0,j=sa[Rank[i]-1];str[i+k]==str[j+k];k++);
}
void ST(int n)
{
    int i,j;
    for(i=1;i<=n;i++)MN[i][0]=height[i];
    for(j=1;(1<<j)<=n;j++)
    {
        for(i=1;i+(1<<j)-1<=n;i++)
        {
            MN[i][j]=min(MN[i][j-1],MN[i+(1<<(j-1))][j-1]);
        }
    }
}
void Insert(int x,int &y,int l,int r,int I)
{
    y=++SIZE;
    sum[y]=sum[x]+1;
    if(l==r)return;
    int mid=(l+r)/2;
    ls[y]=ls[x];rs[y]=rs[x];
    if(I<=mid)Insert(ls[x],ls[y],l,mid,I);
    else Insert(rs[x],rs[y],mid+1,r,I);
}
int Query(int x,int y,int l,int r,int ql,int qr)
{
    if(ql<=l&&r<=qr)return sum[y]-sum[x];
    int mid=(l+r)/2;
    if(qr<=mid)return Query(ls[x],ls[y],l,mid,ql,qr);
    else if(ql>mid)return Query(rs[x],rs[y],mid+1,r,ql,qr);
    else return Query(ls[x],ls[y],l,mid,ql,mid)+Query(rs[x],rs[y],mid+1,r,mid+1,qr);
}
int main()
{
    freopen("heoi2016_str.in","r",stdin);
    freopen("heoi2016_str.out","w",stdout);
    int n,m,lstr,i,s1,s2,s3,s4,wz,l,r,ans,j,mid,last,next,MID,wz1,wz2;
    n=read();m=read();
    scanf("\n%s",str);
    lstr=strlen(str);
    str[lstr+1]=0;
    DA(str,sa,lstr+1,256);
    calheight(lstr);
    ST(lstr);
    SIZE=0;
    for(i=lstr;i>=1;i--)Rank[i]=Rank[i-1];
    for(i=1;i<=n;i++)Insert(root[i-1],root[i],1,n,Rank[i]);
    for(i=1;i<=m;i++)
    {
        s1=read();s2=read();s3=read();s4=read();
        wz=Rank[s3];
        l=1;r=min(s2-s1+1,s4-s3+1);ans=0;
        while(l<=r)
        {
            mid=(l+r)/2;
            wz1=wz2=wz;MID=mid;
            for(j=16;j>=0;j--)if(wz1>=(1<<j)&&MN[wz1-(1<<j)+1][j]>=MID)wz1-=(1<<j);
            for(j=16;j>=0;j--)if(wz2+(1<<j)<=n&&MN[wz2+1][j]>=MID)wz2+=(1<<j);
            if(Query(root[s1-1],root[s2-mid+1],1,n,wz1,wz2)>0){ans=mid;l=mid+1;}
            else r=mid-1;
        }
        printf("%d\n",ans);
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}