BZOJ3879: SvT（后缀数组）

传送门

题解：
后缀自动机+虚树应该是一个挺显然的做法， 不过常数太大。
注意后缀数组两个位置间的height取$\min$与建虚树等价，进一步，两个字符串的$lcp$是后缀数组上height取$\min$，我们按照在后缀数组上相邻的位置排序，建一个新$height$数组，就相当与原来的虚树，单调栈处理出左右比他小的第一个位置，就可以直接统计贡献了。

#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');
    }
}