Bzoj 4556: [Tjoi2016&Heoi2016]字符串

4556: [Tjoi2016&Heoi2016]字符串

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 177  Solved: 92
[Submit][Status][Discuss]

Description

佳媛姐姐过生日的时候,她的小伙伴从某东上买了一个生日礼物。生日礼物放在一个神奇的箱子中。箱子外边写了
一个长为n的字符串s,和m个问题。佳媛姐姐必须正确回答这m个问题,才能打开箱子拿到礼物,升职加薪,出任CE
O,嫁给高富帅,走上人生巅峰。每个问题均有a,b,c,d四个参数,问你子串s[a..b]的所有子串和s[c..d]的最长公
共前缀的长度的最大值是多少?佳媛姐姐并不擅长做这样的问题,所以她向你求助,你该如何帮助她呢?

Input

输入的第一行有两个正整数n,m,分别表示字符串的长度和询问的个数。接下来一行是一个长为n的字符串。接下来
m行,每行有4个数a,b,c,d,表示询问s[a..b]的所有子串和s[c..d]的最长公共前缀的最大值。1<=n,m<=100,000,
字符串中仅有小写英文字母,a<=b,c<=d,1<=a,b,c,d<=n
 

Output

 对于每一次询问,输出答案。

Sample Input

5 5
aaaaa
1 1 1 5
1 5 1 1
2 3 2 3
2 4 2 3
2 3 2 4

Sample Output

1
1
2
2
2

HINT

 

Source

题解:
后缀数组+二分+主席树+ST表
先将原字符串的Rank[],sa[],height[]用后缀数组求出来。用ST表把height[]的区间最小值求出来。把Rank[]建到主席树上。
对于每一组询问(a,b,c,d),先二分答案。然后去找到(c,d)串在后缀中的位置,并向前向后扩展,用ST表把和(c,d)串公共前缀长度大于mid的范围[L,R]找到。最后只需要在主席树上的root[s1~s2-mid+1]中找到Rank[]在范围[L,R]中的即可。
#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;
}
posted @ 2016-06-11 16:45  微弱的世界  阅读(547)  评论(0编辑  收藏  举报