3879: SvT
Time Limit: 30 Sec Memory Limit: 512 MBSubmit: 546 Solved: 219
[ Submit][ Status][ Discuss]
Description
(我并不想告诉你题目名字是什么鬼)
有一个长度为n的仅包含小写字母的字符串S,下标范围为[1,n].
现在有若干组询问,对于每一个询问,我们给出若干个后缀(以其在S中出现的起始位置来表示),求这些后缀两两之间的LCP(LongestCommonPrefix)的长度之和.一对后缀之间的LCP长度仅统计一遍.
Input
第一行两个正整数n,m,分别表示S的长度以及询问的次数.
接下来一行有一个字符串S.
接下来有m组询问,对于每一组询问,均按照以下格式在一行内给出:
首先是一个整数t,表示共有多少个后缀.接下来t个整数分别表示t个后缀在字符串S中的出现位置.
Output
对于每一组询问,输出一行一个整数,表示该组询问的答案.由于答案可能很大,仅需要输出这个答案对于23333333333333333(一个巨大的质数)取模的余数.
Sample Input
7 3
popoqqq
1 4
2 3 5
4 1 2 5 6
popoqqq
1 4
2 3 5
4 1 2 5 6
Sample Output
0
0
2
Hint
样例解释:
对于询问一,只有一个后缀”oqqq”,因此答案为0.
对于询问二,有两个后缀”poqqq”以及”qqq”,两个后缀之间的LCP为0,因此答案为0.
对于询问三,有四个后缀”popoqqq”,”opoqqq”,”qqq”,”qq”,其中只有”qqq”,”qq”两个后缀之间的LCP不为0,且长度为2,因此答案为2.
对于100%的测试数据,有S<=5*10^5,且Σt<=3*10^6.
特别注意:由于另一世界线的某些参数发生了变化,对于一组询问,即使一个后缀出现了多次,也仅算一次.
HINT
Source
题解:后缀数组+单调栈
每次将询问按照rank排序,然后用单调栈维护区间height的最小值,计算答案即可。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 500003
#define p1 23333333333333333LL
#define LL long long
using namespace std;
int n,m,p,l[N],cnt[N],top,stack[N],b[N];
int a[N],v[N],xx[N],yy[N],*x,*y,sa[N],rank[N],height[N],st[20][N];
char s[N];
struct data{
int rk,pos,val;
}c[N];
int cmp(int i,int j,int l)
{
return y[i]==y[j]&&(i+l>n?-1:y[i+l])==(j+l>n?-1:y[j+l]);
}
int cmp1(data a,data b)
{
return a.rk<b.rk;
}
void get_SA()
{
x=xx; y=yy;
memset(v,0,sizeof(v));
int m1=30;
for (int i=1;i<=n;i++) v[x[i]=a[i]]++;
for (int i=1;i<=m1;i++) v[i]+=v[i-1];
for (int i=n;i>=1;i--) sa[v[x[i]]--]=i;
for (int k=1;k<=n;k<<=1) {
p=0;
for (int i=n-k+1;i<=n;i++) y[++p]=i;
for (int i=1;i<=n;i++)
if (sa[i]>k) y[++p]=sa[i]-k;
for (int i=1;i<=m1;i++) v[i]=0;
for (int i=1;i<=n;i++) v[x[y[i]]]++;
for (int i=1;i<=m1;i++) v[i]+=v[i-1];
for (int i=n;i>=1;i--) sa[v[x[y[i]]]--]=y[i];
swap(x,y); p=2; x[sa[1]]=1;
for (int i=2;i<=n;i++)
x[sa[i]]=cmp(sa[i],sa[i-1],k)?p-1:p++;
if (p>n) break;
m1=p+1;
}
for (int i=1;i<=n;i++) rank[sa[i]]=i;
p=0;
for (int i=1;i<=n;i++) {
if (rank[i]==1) continue;
int j=sa[rank[i]-1];
while (i+p<=n&&j+p<=n&&a[i+p]==a[j+p]) p++;
height[rank[i]]=p;
p=max(p-1,0);
}
for (int i=1;i<=n;i++) st[0][i]=height[i];
for (int i=1;i<=17;i++)
for (int j=1;j<=n;j++)
if (j+(1<<i)-1<=n)
st[i][j]=min(st[i-1][j],st[i-1][j+(1<<(i-1))]);
int j=0;
for (int i=1;i<=n;i++) {
if (i>=(1<<(j+1))) j++;
l[i]=j;
}
}
int calc(int x,int y)
{
int k=l[y-x]; x++;
return min(st[k][x],st[k][y-(1<<k)+1]);
}
int main()
{
freopen("a.in","r",stdin);
freopen("my.out","w",stdout);
scanf("%d%d",&n,&m);
scanf("%s",s+1);
for (int i=1;i<=n;i++) a[i]=s[i]-'a';
get_SA();
//for (int i=1;i<=n;i++) cout<<sa[i]<<" ";
//cout<<endl;
//for (int i=1;i<=n;i++) cout<<rank[i]<<" ";
//cout<<endl;
//for (int i=1;i<=n;i++) cout<<height[i]<<" ";
// cout<<endl;
for (int i=1;i<=m;i++) {
int k; scanf("%d",&k);
for (int j=1;j<=k;j++) scanf("%d",&b[j]),b[j]=rank[b[j]];
sort(b+1,b+k+1);
k=unique(b+1,b+k+1)-b-1;
for (int j=1;j<=k;j++)
c[j].rk=b[j];
LL sum=0; LL ans=0; int top=0;
for (int j=2;j<=k;j++)
c[j].val=calc(c[j-1].rk,c[j].rk);
//for (int j=1;j<=k;j++) cout<<c[j].pos<<" "<<c[j].val<<endl;
for (int j=2;j<=k;j++) {
int num=0;
while (top&&c[j].val<stack[top]) {
sum+=(LL)(c[j].val-stack[top])*(LL)cnt[top]%p1;
sum=(sum%p1+p1)%p1;
num+=cnt[top];
top--;
}
stack[++top]=c[j].val;
cnt[top]=num+1; sum+=c[j].val;
ans+=(LL)sum%p1; ans=(ans%p1+p1)%p1;
}
printf("%I64d\n",ans);
}
}