UVA 10829 L-Gap Substrings（后缀数组好题）

题目大意：如果一个字符串它的形式是 UGU，U不为空，那么它就是 G 字符串，现在给你一个字符串，问你它的 L 字符串的个数为多少？
思路：很容易想到用后缀数组做，关键是时间复杂度的问题，普通枚举是 n^2，会爆时间。这里考虑和上一题POJ 3693 一样的思路，
枚举u的长度 u_len,然后每个 u_len ，可以把字符串分为len/u_len组，每组首字母位置为0、u_len、2*u_len...考虑U有且只包含一个位置的情况，
这里的U我们算左边的，考虑包含位置 i*u_len 如果我们知道这其中的某个位置，它和后面的+u_len+g的公共前缀最长为 w，那么就是 ans += （w-u_len+1）。
现在就是怎么求最长的问题，先是算 lcp(i*u_len,i*u_len+u_len+g)，这个是后面的最长，当然前面还有可能存在，再反着算 lcp2（i*u_len-1,i*u_len+u_len+g-1），
另外注意，这两段在算的时候，有可能 lcp 很长，由于枚举的时候不能重复算，算后面时最大是 u_len，算前面最大是 u_len-1，取小即可。和poj那道一样，时间复杂度O（n*logn）。
自己做的时候，竟然没想到枚举 U 的长度，而是枚举 G 的起始位置，发现时间复杂度太高了，唉，失败啊。。。= = 

代码如下：

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;

const int MAXN = 55555<<1;

char str[MAXN];

int sa[MAXN],rank[MAXN],height[MAXN];

int t[MAXN],t2[MAXN],c[MAXN];

void cal_height(int n)
{
    for(int i = 1;i <= n;i++) rank[sa[i]] = i;
    int k = 0;
    for(int i = 0;i < n;i++)
    {
        if(k) k--;
        int j = sa[rank[i]-1];
        while(str[i+k] == str[j+k]) k++;
        height[rank[i]] = k;
    }
}

void da(int n,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] = 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 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]] == y[sa[i-1]] && y[sa[i]+k] == y[sa[i-1]+k] ? p-1 : p++;
        if(p >= n) break;
        m = p;
    }
    cal_height(n-1);
}

int d[MAXN][20];

void rmq_init(int n)
{
    for(int i = 1;i <= n;i++)
        d[i][0] = height[i];
    for(int j = 1;(1<<j) <= n;j++)
        for(int i = 1;i+(1<<j) <= n;i++)
            d[i][j] = min(d[i][j-1],d[i+(1<<(j-1))][j-1]);
}

int rmq(int a,int b)
{
    int k = 0;
    while((1<<(k+1)) < (b-a+1)) k++;
    return min(d[a][k],d[b-(1<<k)+1][k]);
}

int lcp(int a,int b)
{
    a = rank[a],b =rank[b];
    if(a > b) swap(a,b);
    return rmq(a+1,b);
}

int main()
{
    int _;
    scanf("%d",&_);
    for(int cas = 1;cas <= _;cas++)
    {
        int L;
        scanf("%d%s",&L,str);
        int len = strlen(str);
        str[len] = 127;
        for(int i = 0;i < len;i++)
            str[len+i+1] = str[len-i-1];
        str[len+len+1] = '\0';
        da(len+len+2,128);
        rmq_init(len+len+1);
        int ans = 0;
        //puts(str);
        for(int u_len = 1;u_len < len/2;u_len++)
        {
            for(int i = 0;i < len;i += u_len)
            {
                int j = i+u_len+L;
                int sum = 0;
                if(j < len) sum += min(lcp(i,j),u_len);
                if(i >= 1) sum += min(lcp(2*len-(i-1),2*len-(j-1)),u_len-1);
                ans += max(0,sum-u_len+1);
                //printf("u_len = %d,i = %d,j = %d,sum = %d\n",u_len,i,j,sum);
            }
        }
        printf("Case %d: %d\n",cas,ans);
    }
    return 0;
}