bzoj 3879: SvT （后缀数组+单调栈）

3879: SvT

Time Limit: 30 Sec  Memory Limit: 512 MB
Submit: 546  Solved: 219
[ Submit][ Status][ Discuss]

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.

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

HINT

Source

By YGY

[ Submit][ Status][ Discuss]

题解：后缀数组+单调栈

每次将询问按照rank排序，然后用单调栈维护区间height的最小值，计算答案即可。

#include  
#include  
#include  
#include  
#include  
#define N 500003  
#define p1 23333333333333333LL  
#define LL long long  
using namespace std;  
int n,m,p,l[N],cnt[N],top,stack[N],b[N];  
int a[N],v[N],xx[N],yy[N],*x,*y,sa[N],rank[N],height[N],st[20][N];  
char s[N];  
struct data{  
    int rk,pos,val;  
}c[N];  
int cmp(int i,int j,int l)  
{  
    return y[i]==y[j]&&(i+l>n?-1:y[i+l])==(j+l>n?-1:y[j+l]);  
}  
int cmp1(data a,data b)  
{  
    return a.rk=1;i--) sa[v[x[i]]--]=i;  
    for (int k=1;k<=n;k<<=1) {  
        p=0;  
        for (int i=n-k+1;i<=n;i++) y[++p]=i;  
        for (int i=1;i<=n;i++)  
         if (sa[i]>k) y[++p]=sa[i]-k;  
        for (int i=1;i<=m1;i++) v[i]=0;  
        for (int i=1;i<=n;i++) v[x[y[i]]]++;  
        for (int i=1;i<=m1;i++) v[i]+=v[i-1];  
        for (int i=n;i>=1;i--) sa[v[x[y[i]]]--]=y[i];  
        swap(x,y); p=2; x[sa[1]]=1;  
        for (int i=2;i<=n;i++)  
         x[sa[i]]=cmp(sa[i],sa[i-1],k)?p-1:p++;  
        if (p>n) break;  
        m1=p+1;  
    }  
    for (int i=1;i<=n;i++) rank[sa[i]]=i;  
    p=0;  
    for (int i=1;i<=n;i++) {  
       if (rank[i]==1) continue;  
       int j=sa[rank[i]-1];  
       while (i+p<=n&&j+p<=n&&a[i+p]==a[j+p]) p++;  
       height[rank[i]]=p;  
       p=max(p-1,0);  
    }  
    for (int i=1;i<=n;i++) st[0][i]=height[i];  
    for (int i=1;i<=17;i++)  
     for (int j=1;j<=n;j++)   
      if (j+(1<=(1<<(j+1))) j++;  
        l[i]=j;  
    }  
}  
int calc(int x,int y)  
{  
    int k=l[y-x]; x++; 
    return min(st[k][x],st[k][y-(1<