计蒜客 焦作网络赛 String and Times（后缀数组）

题目：给一个字符串问重复出现的次数在[A，B] 区间的子串有多少种。

思路：后缀数组，对于一个height[i],求height[i-1]---height[i]区间有多少个h满足存在大于等于h的连续的最长长度>=A并且<=B的一个序列（h的数量比较少可以单个枚举。。），这里用rmq来实现。然后另外算一下A==1的情况，也就是只出现了一次的子串有多少个（比赛时就是这里写错了，唉，我tmd弱智啊）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=200010;
int t1[maxn],t2[maxn],c[maxn];
bool cmp(int *r,int a,int b,int l)
{
	return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int str[],int sa[],int ra[],int height[],int n,int m)
{
	n++;
	int p,*x=t1,*y=t2;
	for(int i=0;i<m;i++) c[i]=0;
	for(int i=0;i<n;i++) c[x[i]=str[i]]++;
	for(int i=1;i<m;i++) c[i]+=c[i-1];
	for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
	for(int j=1;j<=n;j<<=1)
	{
		p=0;
		for(int i=n-j;i<n;i++) y[p++]=i;
		for(int i=0;i<n;i++)
		if(sa[i]>=j)  y[p++]=sa[i]-j;
		for(int i=0;i<m;i++) c[i]=0;
		for(int i=0;i<n;i++) c[x[y[i]]]++;
		for(int i=1;i<m;i++) c[i]+=c[i-1];
		for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
		swap(x,y);
		p=1;x[sa[0]]=0;
		for(int i=1;i<n;i++)
		x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
		if(p>=n) break;
		m=p;
	}
	int k=0;
	n--;
	for(int i=0;i<=n;i++) ra[sa[i]]=i;
	for(int i=0;i<n;i++)
	{
		if(k) k--;
		int j=sa[ra[i]-1];
		while(str[i+k]==str[j+k])k++;
		height[ra[i]]=k;
	}
}
int m,n,k,t;
int ra[maxn],height[maxn],str[maxn],sa[maxn];
int mm[maxn],best[20][maxn];
void initRMQ(int n)
{
	for(int i=1;i<=n*2;i++)
	for(int j=0;j<19;j++)
	best[j][i]=0;
    mm[0]=-1;
    for(int i=1;i<=n;i++)
        mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
    for(int i=1;i<=n;i++)best[0][i]=height[i];
    for(int i=1;i<=mm[n];i++)
        for(int j=1;j+(1<<i)-1<=n;j++)
        {
            int a=best[i-1][j];
            int b=best[i-1][j+(1<<(i-1))];
            if(a<b) best[i][j]=a;
            else best[i][j]=b;
        }
}
int askRMQ(int a,int b)
{
    if(b>n) return 0;
    int t=mm[b-a+1];
    b-=(1<<t)-1;
    a=best[t][a];b=best[t][b];
    return a<b?a:b;
}

char s[maxn];
int main()
{
    while(~scanf("%s",s))
    {
        int A,B;
        scanf("%d%d",&A,&B);
        n=strlen(s);
        for(int i=0;i<n;i++)
        str[i]=s[i]-'A'+1;
        str[n]=0;
        da(str,sa,ra,height,n,105);
        A--;
        B--;
        initRMQ(n);
        int i=2;
        int AA=A;
        long long ans=0;
        if(B>=1)
        {
        	A=max(A,1);
        	while(i<=n)
	        {
	    		int j=i+A-1,k=i+B-1,tmp=height[i-1];
	    		for(int t=height[i-1]+1;t<=height[i];t++)
		    		if(j<=n&&askRMQ(i,j)>=t&&askRMQ(j+1,k+1)<t)
		           	    ans++;
	        	i++;
	        }
        }
        if(AA==0)
        for(int i=1;i<=n;i++)
        	ans+=1ll*(n-sa[i]-max(height[i],height[i+1]));//求一个串中只出现了一次的字串数量
        printf("%lld\n",ans);
    }
	return 0;
}