bzoj3879 SvT

来源：互联网发布：海外淘宝网编辑：程序博客网时间：2024/06/05 20:36

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

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;#define LL long longconst 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);    }}

0 0