[HEOI2016/TJOI2016]字符串(主席树+二分答案+SA)

include

include

include

include

include

using namespace std;
const int N=101010;
int c[N],x[N],y[N],rk[N],height[N],sa[N],n,m,mn[N][20];
int tot,sum[N*30],ch[N*30][2],root[N];
int t;
char s[N];
void get_sa(){
for(int i=1;i<=n;i++)c[x[i]=s[i]]++;
for(int i=1;i<=m;i++)c[i]+=c[i-1];
for(int i=n;i>=1;i--)sa[c[x[i]]--]=i;
for(int k=1;k<=n;k<<=1){
int num=0;
for(int i=n-k+1;i<=n;i++)y[++num]=i;
for(int i=1;i<=n;i++)if(sa[i]>k)y[++num]=sa[i]-k;
for(int i=1;i<=m;i++)c[i]=0;
for(int i=1;i<=n;i++)c[x[i]]++;
for(int i=1;i<=m;i++)c[i]+=c[i-1];
for(int i=n;i>=1;i--)sa[c[x[y[i]]]--]=y[i],y[i]=0;
for(int i=1;i<=n;i++)swap(x[i],y[i]);
x[sa[1]]=1;num=1;
for(int i=2;i<=n;i++)
x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
if(n==num)break;
m=num;
}
}
void get_height(){
int k=0;
for(int i=1;i<=n;i++)rk[sa[i]]=i;
for(int i=1;i<=n;i++){
if(rk[i]==1)continue;
if(k)k--;
int j=sa[rk[i]-1];
while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k])k++;
height[rk[i]]=k;
}
}
void pre_work(){
for(int i=1;i<=n;i++)mn[i][0]=height[i];
int len=log2(n);
for(int j=1;j<=len;j++)
for(int i=1;i+(1<<j)-1<=n;i++)
mn[i][j]=min(mn[i][j-1],mn[i+(1<<j-1)][j-1]);
}
int get_lcp(int l,int r){
if(l>r)swap(l,r);
l++;
if(l>r)return 1e9;
int len=log2(r-l+1);
return min(mn[l][len],mn[r-(1<<len)+1][len]);
}
void add(int l,int r,int x,int pre,int &now){
now=++tot;
ch[now][0]=ch[pre][0];
ch[now][1]=ch[pre][1];
sum[now]=sum[pre]+1;
if(l==r)return;
int mid=(l+r)>>1;
if(x>mid)add(mid+1,r,x,ch[pre][1],ch[now][1]);
else add(l,mid,x,ch[pre][0],ch[now][0]);
}
int getpre(int l,int r,int x,int pre,int now){
if(sum[now]-sum[pre]==0)return 0;
if(l==r)return l;
int mid=(l+r)>>1;
if(x<=mid)return getpre(l,mid,x,ch[pre][0],ch[now][0]);
int ans=getpre(mid+1,r,x,ch[pre][1],ch[now][1]);
if(ans)return ans;
else return getpre(l,mid,x,ch[pre][0],ch[now][0]);
}
int getsuc(int l,int r,int x,int pre,int now){
if(sum[now]-sum[pre]==0)return 0;
if(l==r)return l;
int mid=(l+r)>>1;
if(x>=mid+1)return getsuc(mid+1,r,x,ch[pre][1],ch[now][1]);
int ans=getsuc(l,mid,x,ch[pre][0],ch[now][0]);
if(ans)return ans;
else return getsuc(mid+1,r,x,ch[pre][1],ch[now][1]);
}
bool check(int l,int r,int x,int pre,int now){
if(l==r)return (sum[now]-sum[pre])>0;
int mid=(l+r)>>1;
if(x>mid)return check(mid+1,r,x,ch[pre][1],ch[now][1]);
else return check(l,mid,x,ch[pre][0],ch[now][0]);
}
bool judge(int l,int r,int L,int x){
if(check(1,n,rk[L],root[l-1],root[r]))return true;
return max(get_lcp(rk[L],getsuc(1,n,rk[L],root[l-1],root[r])),get_lcp(rk[L],getpre(1,n,rk[L],root[l-1],root[r])))>=x;
}
int read(){
int sum=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){sum=sum10+ch-'0';ch=getchar();}
return sum
f;
}
int main(){
n=read();t=read();
scanf("%s",s+1);
m=122;
get_sa();get_height();pre_work();
for(int i=1;i<=n;i++)add(1,n,rk[i],root[i-1],root[i]);
while(t--){
int a=read(),b=read(),c=read(),d=read();
int l=1,r=min(b-a+1,d-c+1);
int ans=0;
while(l<=r){
int mid=(l+r)>>1;
if(judge(a,b-mid+1,c,mid))ans=mid,l=mid+1;
else r=mid-1;
}
printf("%d\n",ans);
}
return 0;
}

猜你喜欢

转载自www.cnblogs.com/Xu-daxia/p/10217088.html
今日推荐