bzoj 3230 相似子串

3230: 相似子串

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 1767  Solved: 438
[Submit][Status][Discuss]

Description

Input

输入第1行，包含3个整数N，Q。Q代表询问组数。
第2行是字符串S。
接下来Q行，每行两个整数i和j。（1≤i≤j）。

Output

输出共Q行，每行一个数表示每组询问的答案。如果不存在第i个子串或第j个子串，则输出-1。

Sample Input

5 3
ababa
3 5
5 9
8 10

Sample Output

18
16
-1

HINT

样例解释

第1组询问：两个子串是“aba”,“ababa”。f = 32 + 32 = 18。

第2组询问：两个子串是“ababa”,“baba”。f = 02 + 42 = 16。

第3组询问：不存在第10个子串。输出-1。

数据范围

N≤100000，Q≤100000，字符串只由小写字母'a'~'z'组成

Source

后缀数组+二分+RMQ

【分析】

不得不说我已经把后缀数组的题写成黑箱题了...

又是一道手玩出来的题...

搞一个原串的后缀数组，再搞一个反串的...

分别搞一下lcp就行了...

不过我还是不太清楚为什么是lower_bound，总感觉是upper_bound-1才对啊...脑洞开了改了一下就A掉了...= =

以上的话当p处理...窝脑子water了...

【代码】

//bzoj 3230 相似子串#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define ll long long#define M(a) memset(a,0,sizeof a)#define fo(i,j,k) for(i=j;i<=k;i++)using namespace std;const int mxn=100005;int m,q;char s[mxn];ll cnt[mxn];ll sum,len,u,v;inline ll read(){    ll x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();    return x*f;}struct node  //大家好我是黑箱 {    int st[mxn][30],rank[mxn],height[mxn];    int a[mxn],b[mxn],x[mxn],y[mxn],sa[mxn];    inline bool comp(int i,int j,int l)    {        return y[i]==y[j]&&(i+l>len?-1:y[i+l])==(j+l>len?-1:y[j+l]);    }    inline void work()    {        int i,j,k,p;m=128;        fo(i,0,m) b[i]=0;        fo(i,1,len) b[x[i]=a[i]]++;        fo(i,1,m) b[i]+=b[i-1];        for(i=len;i>=1;i--) sa[b[x[i]]--]=i;        for(k=1;k<=len;k<<=1)        {            p=0;            fo(i,len-k+1,len) y[++p]=i;            fo(i,1,len) if(sa[i]>k) y[++p]=sa[i]-k;            fo(i,0,m) b[i]=0;            fo(i,1,len) b[x[y[i]]]++;            fo(i,1,m) b[i]+=b[i-1];            for(i=len;i>=1;i--) sa[b[x[y[i]]]--]=y[i];            swap(x,y),p=2,x[sa[1]]=1;            fo(i,2,len)              x[sa[i]]=comp(sa[i-1],sa[i],k)?p-1:p++;            if(p>len) break;            m=p;        }        p=k=0;        fo(i,1,len) rank[sa[i]]=i;        for(i=1;i<=len;height[rank[i++]]=k)          for(k?k--:0,j=sa[rank[i]-1];a[i+k]==a[j+k];k++);    }    inline void init()    {        int i,j;        fo(i,1,len) st[i][0]=height[i];        fo(j,1,28)          fo(i,1,len) if((i+(1<<j)-1)<=len)            st[i][j]=min(st[i][j-1],st[i+(1<<j-1)][j-1]);    }    inline int lcp(int x,int y)    {        if(x==y) return len-x+1;        x=rank[x],y=rank[y];        if(x>y) swap(x,y);        x++;        int k=0;        while(x+(1<<k+1)<=y) k++;        return min(st[x][k],st[y-(1<<k)+1][k]);    }}A,B;int main(){    int i,j,k;    len=read(),q=read();    scanf("%s",s+1);    fo(i,1,len) A.a[i]=s[i];    fo(i,1,len) B.a[i]=s[len-i+1];    A.work(),A.init();    B.work(),B.init();    sum=len*(len+1)/2;    fo(i,1,len) sum-=A.height[i];    fo(i,1,len)      cnt[i]=cnt[i-1]+len-A.sa[i]+1-A.height[i];    while(q--)    {        u=read(),v=read();        if(v>sum)        {            printf("-1\n");            continue;        }        int Tu=lower_bound(cnt+1,cnt+len+1,u)-cnt;        int Tv=lower_bound(cnt+1,cnt+len+1,v)-cnt;        int tu=u-cnt[Tu-1]+A.sa[Tu]+A.height[Tu]-1;        int tv=v-cnt[Tv-1]+A.sa[Tv]+A.height[Tv]-1;        ll start=min(A.lcp(A.sa[Tu],A.sa[Tv]),min(tu-A.sa[Tu]+1,tv-A.sa[Tv]+1));        ll end=min(B.lcp(len-tu+1,len-tv+1),min(tu-A.sa[Tu]+1,tv-A.sa[Tv]+1));        printf("%lld\n",start*start+end*end);    }    return 0;}