UVA 10829 L-Gap Substrings (后缀数组+RMQ)

先枚举U的长度，然后枚举坐端点，左断点向左向右扩展，如果长度大于L，那么lcp-L+1就是该点的UVU串数量。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100000+5;
int g;
struct SuffixArray{
    int s[maxn];
    int sa[maxn];
    int rank[maxn];
    int height[maxn];
    int t[maxn],t2[maxn],c[maxn],n;
    int d[maxn][20];
    void clear() {n=0;memset(sa,0,sizeof(sa));}
    void build_sa(int m)
    {
        int *x=t,*y=t2;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[i]=s[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 k=1;k<=n;k<<=1)
        {
            int p=0;
            for(int i=n-k;i<n;i++) y[p++]=i;
            for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
            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]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
            if(p>=n) break;
            m=p;
        }
    }
    
    void build_height(){
        int k=0;
        for(int i=0;i<n;i++) rank[sa[i]]=i;
        for(int i=0;i<n;i++){
            if(k) k--;
            int j=sa[rank[i]-1];
            while(s[i+k]==s[j+k]) k++;
            height[rank[i]]=k;
        }
    }
    
    void RMQ_init(){
        for(int i=0;i<n;i++) d[i][0]=height[i];
        for(int j=1;(1<<j)<=n;j++)
            for(int i=0;i+(1<<j)-1<n;i++)
                d[i][j]=min(d[i][j-1],d[i+(1<<(j-1))][j-1]);
    }
    
    int RMQ(int L,int R){
        if(L>=R) swap(L,R);L++;
        int k=0;
        while((1<<(k+1))<=R-L+1) k++;
        return min(d[L][k],d[R-(1<<k)+1][k]);
    }
    
    long long count(int len){
        long long ans=0;
        for(int l=1;l<=len/2;l++)
        {
            for(int j=0;j<len;j+=l)
            {
                int lcp=0;
                if(j+l+g<len) lcp+=min(l,RMQ(rank[j],rank[j+l+g]));
                if(n-j-l-g-1>=0) lcp+=min(l-1,RMQ(rank[n-j-1],rank[n-j-l-g-1]));
                ans+=max(0,lcp-l+1);
            }
        }
        return ans;
    }
};

#define rep(i,a,b) for(int i=(a);i<(b);i++)
SuffixArray sa;
char s[maxn];
int main()
{
    int t,kase=0;
    scanf("%d",&t);
    while(t--)
    {
        sa.clear();
        scanf("%d%s",&g,s);
        int len=(int)strlen(s);
        rep(i,0,len) sa.s[sa.n++]=s[i]-'a'+1;sa.s[sa.n++]=27;
        reverse(s,s+len);
        rep(i,0,len) sa.s[sa.n++]=s[i]-'a'+1;sa.s[sa.n++]=0;
        sa.build_sa(30);
        sa.build_height();
        sa.RMQ_init();
        printf("Case %d: %lld\n",++kase,sa.count(len));
    }
    return 0;
}