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
题解:
网上都说是后缀树,然而本蒟蒻并不会后缀树。
所以只能打后缀数组。
如果暴力求lcp的话,时间复杂度为n^2,好像太高了。
所以我们要优化,用单调栈来维护。
维护一个不下降的序列,时间复杂就低了。
具体操作见代码。
什么?!这题要模233333333?然而我没模也过了。(其实是忘记模了。。)
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stack>
#include<iostream>
using namespace std;
const int N=500003;
#define ll long long
int n,m;
int ww[N],wy[N],wx[N],wv[N];
int cmp(int *r,int x,int x1,int ln)
{
return r[x]==r[x1]&&r[x+ln]==r[x1+ln];
}
void da(int *r,int *sa,int n,int m)
{
int *x=wx,*y=wy,*t,i,j,p;
for(i=0;i<m;i++) ww[i]=0;
for(i=0;i<n;i++) ww[x[i]=r[i]]++;
for(i=1;i<m;i++) ww[i]+=ww[i-1];
for(i=n-1;i>=0;i--) sa[--ww[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++) ww[i]=0;
for(i=0;i<n;i++) ww[wv[i]]++;
for(i=1;i<m;i++) ww[i]+=ww[i-1];
for(i=n-1;i>=0;i--) sa[--ww[wv[i]]]=y[i];
for(t=x,x=y,y=t,i=1,x[sa[0]]=0,p=1;i<n;i++)
x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
}
}
int rank[N],h[N];
void get_h(int *r,int *sa,int n)
{
int i,j,k=0;
for(i=1;i<=n;i++) rank[sa[i]]=i;
for(i=0;i<n;h[rank[i++]]=k)
for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
return;
}
char s[N];
int r[N],sa[N];
int f[N][30],q[N<<3];
bool cmp1(int x,int y)
{
return rank[x-1]<rank[y-1];
}
int lcp(int x,int y)
{
if(x>y) swap(x,y);
int t=log2(y-x);
return min(f[x+1][t],f[y-(1<<t)+1][t]);
}
stack<int>S;
int siz[N];
ll tmp[N];
int main()
{
scanf("%d%d",&n,&m);
scanf("%s",s);
for(int i=0;i<n;i++) r[i]=s[i]-'a'+1;
da(r,sa,n+1,130);
get_h(r,sa,n);
//for(int i=1;i<=n;i++) printf("%d ",h[i]);
for(int i=1;i<=n;i++) f[i][0]=h[i];
for(int i=1;(1<<i)<=n;i++)
for(int j=1;j+(1<<i)-1<=n;j++)
f[j][i]=min(f[j][i-1],f[j+(1<<i-1)][i-1]);
while(m--)
{
int t;
scanf("%d",&t);
for(int i=1;i<=t;i++) scanf("%d",&q[i]);
sort(q+1,q+1+t,cmp1);
t=unique(q+1,q+1+t)-q-1;//去重
ll ans=0,sum=0;
while(!S.empty())S.pop();
for(int i=2;i<=t;i++)
{
siz[i]=1;
tmp[i]=lcp(rank[q[i-1]-1],rank[q[i]-1]);
while(!S.empty()&&tmp[S.top()]>tmp[i])
sum-=(ll)(tmp[S.top()]-tmp[i])*siz[S.top()],
siz[i]+=siz[S.top()],
S.pop();
sum+=tmp[i];ans+=sum;
S.push(i);
}
printf("%lld\n",ans);
}
}