HDU 6194

首先处理出来Height数组，然后初始化ST表，那么我们从头开始RMQ查询长度为k的区间，那么区间最小值那个前缀至少出现了k次，然后我们将区间左右扩展，目的就是为了去除超过k次的，然后减掉多出来的就可以。

#include<bits/stdc++.h>using namespace std;#define N 500000int st[N][50],n,rk[N],sa[N],height[N],tmp[N],cnt[N];char s[N];void suffixarray(int n,int m) {  //n为字符串长度，m为基数排序的阙值    int i,j,k;n++;    for(i=0;i<n*2+5;i++) rk[i]=sa[i]=height[i]=tmp[i]=0;    for(i=0;i<m;i++) cnt[i]=0;    for(i=0;i<n;i++) cnt[rk[i]=s[i]]++;    for(i=1;i<m;i++) cnt[i]+=cnt[i-1];    for(i=0;i<n;i++) sa[--cnt[rk[i]]]=i;  //  for(i=0;i<n;i++)  //      cout<<"sa="<<sa[i]<<' ';  //  cout<<endl;    for(k=1;k<=n;k<<=1)    {        for(i=0;i<n;i++)        {            j=sa[i]-k;            if(j<0) j+=n;            tmp[cnt[rk[j]]++]=j;        }        sa[tmp[cnt[0]=0]]=j=0;        for(i=1;i<n;i++)        {            if(rk[tmp[i]]!=rk[tmp[i-1]]||rk[tmp[i]+k]!=rk[tmp[i-1]+k]) cnt[++j]=i;            sa[tmp[i]]=j;        }        memcpy(rk,sa,n*sizeof(int));        memcpy(sa,tmp,n*sizeof(int));        if(j>=n-1) break;    }    for(j=rk[height[i=k=0]=0];i<n-1;i++,k++)    {        while(~k&&s[i]!=s[sa[j-1]+k]) height[j]=k--,j=rk[sa[j]+1];    }}void RMQ_init(){   // cout<<"n="<<n<<endl;    int i;    for(i=0;i<=n;i++)        st[i][0]=height[i];    for(i=1;(1<<i)<=n;i++)    {      // cout<<i<<endl;        for(int j=2;j+(1<<i)-1<=n;j++)        {            st[j][i]=min(st[j][i-1],st[j+(1<<(i-1))][i-1]);        }    }//    for(int i=1;i<=n;i++)//        cout<<height[i]<<' '<<s+sa[i]<<endl;//    for(int i=2;i<=n;i++)//    {//        for(int j=0;j<=5;j++)//            cout<<st[i][j]<<' ';//        cout<<endl;//    }}int RMQ(int l,int r){    if(l==r) return strlen(s+sa[l]);    l++;    r++;    int len=log2(r-l);  //  cout<<len<<endl;    return min(st[l][len],st[r-(1<<len)][len]);}void init(){    n=strlen(s);    suffixarray(n,500);    RMQ_init();}int main(){    //test    int t;    cin>>t;    while(t--)    {        int k;        scanf("%d",&k);        scanf("%s",s);        init();        int ans=0;        for(int i=1;i+k-1<=n;i++)        {            int l=i,r=i+k-1;            int len=RMQ(l,r);            int more=max((l>1?RMQ(l-1,r):0),(r<n?RMQ(l,r+1):0));            //cout<<"len="<<len<<" more="<<more<<endl;            ans+=max(0,(len-more));        }        cout<<ans<<endl;    }}