HDU_6194 后缀数组+RMQ

好绝望的。。想了五个多小时，最后还是没A。。。赛后看了下后缀数组瞬间就有了思路。。。不过因为太菜，想了将近两个小时才吧这个题干掉。

首先，应当认为，后缀数组的定义是，某字符串S的所有后缀按照字典序有小到大的顺序排列（使用下标表示后缀）。因为具体过程没太看懂，但是参见刘汝佳蓝书《算法竞赛黑暗圣典》可以得到一个聪明的NLOGN的神器算法。不过这个不太重要。

之后还可以通过他在LCP问题中提到的RANK，height数组相关算法，处理出来height数组，之后其他的可以扔掉。

《黑暗圣典》中定义了height数组，height[k]的含义是，第rank[i]数组和rank[i]-1之间的最长公共前缀的长度。。。基于这个我们可以看到一些规则。

首先height数组的定义具有传递性，很容易想到的就是。。。出现且仅出现M次可以被简单的定义为，传递且仅能够被传递M次。。。

于是。。。我们有了如下算法.

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
const long long MAXN=;
const long long INF=1E8+;
char s[MAXN];
//long long dp[MAXN];
long long sa[MAXN],t[MAXN],t2[MAXN],c[MAXN],n,len;
long long r1ank[MAXN],height[MAXN];
long long d[MAXN][];
void RMQ_init()
{
    for(int i=;i<n;++i)d[i][]=height[i];
    for(int j=;(<<j)<=n;j++)
        for(int i=;i+(<<j)-<n;i++)
            d[i][j]=min(d[i][j-],d[i+(<<(j-))][j-]);
}
long long RMQ(int a,int b)
{
    int k=;
    while((<<(k+))<=b-a+)k++;
    return min(d[a][k],d[b-(<<k)+][k]);
}

int m;
void build_sa(int m)
{
    long long i,*x=t,*y=t2;
    for( i=;i<m;++i)c[i]=;
    for( i=;i<n;++i)c[x[i]=s[i]]++;
    for( i=;i<m;++i)c[i]+=c[i-];
    for( i=n-;i>=;i--)sa[--c[x[i]]]=i;
    for( int k=;k<= n ;k*=)
    {
        int p=;
        for(i=n-k;i<n;++i)y[p++]=i;
        for(i=;i<n;++i)if(sa[i]>=k)y[p++]=sa[i]-k;

        for( i=;i<m;++i)c[i]=;
        for( i=;i<n;++i)c[x[y[i]]]++;
        for( i=;i<m;++i)c[i]+=c[i-];
        for( i=n-;i>=;i--)sa[--c[x[y[i]]]]=y[i];

        swap(x,y);
        p=;x[sa[]]=;
        for( i=;i<n;++i)
        {
            x[sa[i]]=y[sa[i-]]==y[sa[i]]&&y[sa[i-]+k]==y[sa[i]+k]?p-:p++;
        }if(p>=n)break;
            m=p;
    }
}
void getHeight()
{
    int i,j,k=;
    for(int i=;i<n;++i)r1ank[sa[i]]=i;
    for(int i=;i<n;++i)
    {
        if(k)k--;
        int j=sa[r1ank[i]-];
        while(s[j+k]==s[i+k])k++;
        height[r1ank[i]]=k;
    }
    height[n]=;
}

int main()
{
    int t;cin>>t;

    while(t--)
    {
//        memset(height,0,sizeof(height));
        scanf("%d %s",&m,s);
        n=strlen(s);
        len=n;n+=;;
        build_sa();
        getHeight();
        RMQ_init();long long ans=;
        for(int i=;i<n;++i)
        {
            if(m==)
            {
                ans+=len-sa[i]-max(height[i],height[i+]);
                continue;
            }
            int a=i+;int b=i+m-;
            long long limit=RMQ(a,b);
            if(n>b&&limit>max(height[i],height[b+]))ans+=limit-max(height[i],height[b+]);

        }
        cout<<ans<<"\n";
    }

}