3230: 相似子串
Time Limit: 20 Sec Memory Limit: 128 MBSubmit: 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
ababa
3 5
5 9
8 10
Sample Output
18
16
-1
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
【分析】
不得不说我已经把后缀数组的题写成黑箱题了...
又是一道手玩出来的题...
搞一个原串的后缀数组,再搞一个反串的...
分别搞一下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;
}