bzoj3879 SvT

这题是【AHOI2013】bzoj3238 差异的加强版。
但是也差不多，对每个询问只把询问到的那些位置拉出来用单调栈维护就行了，用ST表维护区间最小值可以做到$O(n\log n+\sum t)$。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define LL long long
const int maxn=500010;
char s[maxn];
int sa[maxn],rank[maxn],height[maxn],cnt[maxn],f[maxn],
a[3000010],sta[maxn],mn[maxn][25],log[maxn],num[maxn],
n,q,m=26;
int get(int l,int r)
{
    int k=log[r-l+1];
    return min(mn[l][k],mn[r-(1<<k)+1][k]);
}
int main()
{
    LL now,ans;
    int x,p,top,tot;
    scanf("%d%d",&n,&q);
    scanf("%s",s+1);
    for (int i=1;i<=n;i++) cnt[rank[i]=f[i]=s[i]-'a'+1]++;
    for (int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
    for (int i=n;i;i--) sa[cnt[f[i]]--]=i;
    for (int k=1;k<=n;k<<=1)
    {
        p=0;
        for (int i=n-k+1;i<=n;i++) f[++p]=i;
        for (int i=1;i<=n;i++)
            if (sa[i]>k) f[++p]=sa[i]-k;
        for (int i=1;i<=m;i++) cnt[i]=0;
        for (int i=1;i<=n;i++) cnt[rank[f[i]]]++;
        for (int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
        for (int i=n;i;i--) sa[cnt[rank[f[i]]]--]=f[i];
        for (int i=1;i<=n;i++) f[i]=rank[i];
        rank[sa[1]]=1;
        for (int i=2;i<=n;i++)
            if (f[sa[i]]!=f[sa[i-1]]||f[sa[i]+k]!=f[sa[i-1]+k])
                rank[sa[i]]=rank[sa[i-1]]+1;
            else
                rank[sa[i]]=rank[sa[i-1]];
        m=rank[sa[n]];
        if (m>=n) break;
    }
    for (int i=1;i<=n;i++)
    {
        height[rank[i]]=height[rank[i-1]];
        if (height[rank[i]]) height[rank[i]]--;
        while (s[i+height[rank[i]]]==s[sa[rank[i]-1]+height[rank[i]]]) height[rank[i]]++;
    }
    for (int i=1;i<=n;i++) mn[i][0]=height[i];
    for (int k=1;(1<<k)<=n;k++) log[1<<k]=k;
    for (int i=3;i<=n;i++) if (!log[i]) log[i]=log[i-1];
    for (int k=1;(1<<k)<=n;k++)
        for (int i=1;i+(1<<k)-1<=n;i++)
            mn[i][k]=min(mn[i][k-1],mn[i+(1<<k-1)][k-1]);
    while (q--)
    {
        scanf("%d",&m);
        for (int i=1;i<=m;i++)
        {
            scanf("%d",&x);
            a[i]=rank[x];
        }
        sort(a+1,a+m+1);
        m=unique(a+1,a+m+1)-a-1;
        now=sta[top=1]=n-sa[a[1]]+1;
        ans=0;
        num[1]=1;
        for (int i=2;i<=m;i++)
        {
            x=get(a[i-1]+1,a[i]);
            tot=0;
            while (top&&x<=sta[top])
            {
                now-=(LL)num[top]*sta[top];
                tot+=num[top];
                top--;
            }
            now+=(LL)x*tot;
            ans+=now;
            sta[++top]=x;
            num[top]=tot;
            sta[++top]=n-sa[a[i]]+1;
            num[top]=1;
            now+=sta[top];
        }
        printf("%lld\n",ans);
    }
}