题解:
后缀自动机+虚树应该是一个挺显然的做法, 不过常数太大。
注意后缀数组两个位置间的height取
min
min
与建虚树等价,进一步,两个字符串的
lcp
l
c
p
是后缀数组上height取
min
min
,我们按照在后缀数组上相邻的位置排序,建一个新
height
h
e
i
g
h
t
数组,就相当与原来的虚树,单调栈处理出左右比他小的第一个位置,就可以直接统计贡献了。
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair <int,int> pii;
inline int rd() {
char ch=getchar(); int i=0,f=1;
while(!isdigit(ch)) {if(ch=='-')f=-1; ch=getchar();}
while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=getchar();}
return i*f;
}
inline void W(LL x) {
static int buf[50];
if(!x) {putchar('0'); return;}
if(x<0) {putchar('-'); x=-x;}
while(x) {buf[++buf[0]]=x%10; x/=10;}
while(buf[0]) {putchar(buf[buf[0]--]+'0');}
}
const int N=1e6+50;
int n,m,q,tp[N],tp2[N],*rk=tp,*sa2=tp2,sa[N],h[20][N],Lg[N];
char ch[N];
inline void Rsort() {
static int c[N];
for(int i=1;i<=m;i++) c[i]=0;
for(int i=1;i<=n;i++) c[rk[i]]++;
for(int i=1;i<=m;i++) c[i]+=c[i-1];
for(int i=n;i>=1;i--) sa[c[rk[sa2[i]]]--]=sa2[i];
}
inline void build_sa() {
for(int i=1;i<=n;i++) rk[i]=ch[i]-'a'+1, sa2[i]=i;
m=26, Rsort();
for(int w=1,p=0;m!=n;w<<=1) {
for(int i=n-w+1;i<=n;i++) sa2[++p]=i;
for(int i=1;i<=n;i++) if(sa[i]>w) sa2[++p]=sa[i]-w;
Rsort(); swap(rk,sa2); rk[sa[1]]=p=1;
for(int i=2;i<=n;i++)
rk[sa[i]]=(sa2[sa[i]]==sa2[sa[i-1]] && sa2[sa[i]+w]==sa2[sa[i-1]+w]) ? p : ++p;
m=p; p=0;
}
for(int i=1,k=0,j;i<=n;h[0][rk[i++]]=k)
for(k?k--:0, j=sa[rk[i]-1]; ch[i+k]==ch[j+k]; k++);
Lg[1]=0;
for(int i=2;i<=n;i++) Lg[i]=Lg[i>>1]+1;
for(int i=1;i<=Lg[n];++i)
for(int j=1; j+(1<<i)-1<=n; ++j)
h[i][j]=min(h[i-1][j],h[i-1][j+(1<<(i-1))]);
}
inline bool cmp_rk(const int &a,const int &b) {return rk[a]<rk[b];}
inline int lcp(int l,int r) {
if(l==r) return n-l+1;
l=rk[l]+1, r=rk[r];
int o=Lg[r-l+1];
return min(h[o][l],h[o][r-(1<<o)+1]);
}
int main() {
n=rd(), q=rd();
scanf("%s",ch+1);
build_sa();
for(int i=1;i<=q;i++) {
static int a[N], hi[N], L[N], R[N];
int t=rd();
for(int i=1;i<=t;i++) a[i]=rd();
sort(a+1,a+t+1,cmp_rk);
t=unique(a+1,a+t+1)-a-1;
for(int i=1;i<t;i++) hi[i]=lcp(a[i],a[i+1]);
static pii stk[N]; static int tl;
tl=0;
for(int i=1;i<t;i++) {
while(tl && stk[tl].first>hi[i]) --tl;
L[i]=tl ? (stk[tl].second+1) : 1;
stk[++tl]=pii(hi[i],i);
}
tl=0;
for(int i=t-1;i>=1;i--) {
while(tl && stk[tl].first>=hi[i]) --tl;
R[i]=tl ? (stk[tl].second-1) : (t-1);
stk[++tl]=(pii(hi[i],i));
}
LL ans=0;
for(int i=1;i<t;i++)
ans+=(LL)hi[i]*(i-L[i]+1)*(R[i]-i+1);
W(ans); putchar('\n');
}
}