bzoj3879: SvT【后缀数组+稀疏表+单调栈】

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

Sample Output

0

0

2

Hint

样例解释:

对于询问一,只有一个后缀”oqqq”,因此答案为0.

对于询问二,有两个后缀”poqqq”以及”qqq”,两个后缀之间的LCP为0,因此答案为0.

对于询问三,有四个后缀”popoqqq”,”opoqqq”,”qqq”,”qq”,其中只有”qqq”,”qq”两个后缀之间的LCP不为0,且长度为2,因此答案为2.

对于100%的测试数据,有S<=5*10^5,且Σt<=3*10^6.

特别注意:由于另一世界线的某些参数发生了变化,对于一组询问,即使一个后缀出现了多次,也仅算一次.

解题思路：

求出后缀数组。
每次询问把位置按rank排序，结合ST表处理相邻询问位置的height数组。
再用单调栈求出每个height作为最小值的区间即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
void print(ll x)
{
    if(x<0){putchar('-');x=-x;}
    if(x>9)print(x/10);
    putchar('0'+x%10);
}
const int N=500005;
int n,q;
int m,rank[N],sa[N],tp[N],c[N],height[N],Log[N],mn[20][N<<1];
int a[N],h[N],stk[N],L[N],R[N];
char s[N];
inline bool cmp(const int &a,const int &b){return rank[a]<rank[b];}
inline void Rsort()
{
    for(int i=1;i<=m;i++)c[i]=0;
    for(int i=1;i<=n;i++)c[rank[tp[i]]]++;
    for(int i=1;i<=m;i++)c[i]+=c[i-1];
    for(int i=n;i;i--)sa[c[rank[tp[i]]]--]=tp[i];
}
inline void build_SA()
{
    m=26;
    for(int i=1;i<=n;i++)tp[i]=i,rank[i]=s[i]-'a'+1;
    Rsort();
    for(int k=1;k<=n;k<<=1)
    {
        int j=0;
        for(int i=n-k+1;i<=n;i++)tp[++j]=i;
        for(int i=1;i<=n;i++)if(sa[i]>k)tp[++j]=sa[i]-k;
        Rsort();swap(rank,tp);j=rank[sa[1]]=1;
        for(int i=2;i<=n;i++)
            rank[sa[i]]=(tp[sa[i]]==tp[sa[i-1]]&&tp[sa[i]+k]==tp[sa[i-1]+k]?j:++j);
        m=j;
    }
    for(int i=1,k=0,j;i<=n;height[rank[i++]]=k)
        for(k=k?--k:0,j=sa[rank[i]-1];s[i+k]==s[j+k];k++);
}
inline void build_ST()
{
    Log[1]=0;for(int i=2;i<=n;i++)Log[i]=Log[i>>1]+1;
    for(int i=1;i<=n;i++)mn[0][i]=height[i];
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i<=n;i++)
            mn[j][i]=min(mn[j-1][i],mn[j-1][i+(1<<j-1)]);
}
inline int lcp(int i,int j)
{
    if(i==j)return n-i+1;
    int l=rank[i]+1,r=rank[j];
    int t=Log[r-l+1];
    return min(mn[t][l],mn[t][r-(1<<t)+1]);
}
int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),q=getint();
    scanf("%s",s+1);
    build_SA();build_ST();
    while(q--)
    {
        int k=getint(),top;ll ans=0;
        for(int i=1;i<=k;i++)a[i]=getint();
        sort(a+1,a+k+1,cmp),k=unique(a+1,a+k+1)-a-1;
        for(int i=1;i<k;i++)h[i]=lcp(a[i],a[i+1]);
        h[0]=h[k]=-1;
        stk[top=0]=0;
        for(int i=1;i<k;i++)
        {
            while(h[stk[top]]>h[i])top--;
            L[i]=stk[top],stk[++top]=i;
        }
        stk[top=0]=k;
        for(int i=k-1;i;i--)
        {
            while(h[stk[top]]>=h[i])top--;
            R[i]=stk[top],stk[++top]=i;
        }
        for(int i=1;i<k;i++)ans+=1ll*h[i]*(i-L[i])*(R[i]-i);
        print(ans),putchar('\n');
    }
    return 0;
}