luogu P1117( lcp 模板 )

题意：
给你一个字符串，问你有多少个子串，形式是AABB AABBAABB，其中A AA和B BB可以相同。串长 <=30000，对于同一个字符串，不同的拆分属于不同的方案。

思路：对于形式为AABB的字串，我们考虑枚举A和B之间的断点，再使用乘法原理进行计算。

          接下来我们考虑如何计算以某一个位置为开始和某一位置为结尾的AA的方案数。

          首先我们枚举A的长度len，并且每隔 len 设置一个断点。

          可以证明任何一个AA只与两个断点相交。

         然后我们求出这两个断点的最长公共前缀和最长公共后缀，便 可以找到所有合法的子串。

//最长回文子串 , lcp 模板
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef int lint;
const lint maxn = 30015;
typedef int lint;
struct suffix{

    lint c[maxn],sa[maxn],t1[maxn],t2[maxn],m,n,h[maxn],height[maxn],rk[maxn],s[maxn];

    void Build_SA(lint len){
        m = 0;
        lint* x = t1,*y = t2;
        n = len;
        for( lint i = 0;i < n;i++ ) m = max( m,s[i] );
        for( lint i = 0;i <= m;i++ ) c[i] = 0;
        for( lint i = 0;i < n;i++ ) c[ x[i] = s[i] ]++;
        for( lint i = 1;i <= m;i++ ) c[i] += c[i-1];
        for( lint i = n-1;i >= 0;i-- ) {
            sa[ --c[ x[i] ] ] = i;
        }
        for( lint k = 1;k < n;k <<= 1 ){
            lint cnt = 0;
            for( lint i = n - 1;i >= n-k;i-- ) y[cnt++] = i;
            for( lint i = 0;i < n;i++ ) if( sa[i] >= k ) y[cnt++] = sa[i]-k;
            for( lint i = 0;i <= m;i++ ) c[i] = 0;
            for( lint i = 0;i < n;i++ ) c[ x[i] ]++;
            for( lint i = 1;i <= m;i++ ) c[i] += c[i-1];
            for( lint i = cnt-1;i >= 0;i-- ) sa[ --c[ x[ y[i] ] ] ] = y[i];
            swap( x,y );
            lint num = 0;
            x[ sa[0] ] = 0;
            for( lint i = 1;i < n;i++ ){
                if( y[ sa[i-1] ] != y[ sa[i] ] || y[ sa[i-1]+k ] != y[ sa[i]+k ] ){
                    x[ sa[i] ] = ++num;
                }else{
                    x[ sa[i] ] = num;
                }
            }
            if( num == n-1 ) return;
            m = num;
        }
    }
    void getheight(){
        for( lint i = 0;i < n;i++ ){
            rk[ sa[i] ] = i;
        }
        lint cnt = 1;
        h[ sa[0] ] = 0;
        height[ 0 ] = 0;
        for( lint i = 0;i < n;i++ ){
            if(cnt)cnt--;
            while(  rk[i] >= 1 && i + cnt < n &&sa[rk[i]-1]+cnt < n  && s[ sa[rk[i]-1]+cnt ] == s[ i+cnt ] ) cnt++;
            h[ i ] = cnt;
            height[ rk[i] ] = cnt;
        }
        //cout << "debug" << endl;
    }

}g,g2;
lint rmq[maxn][20],lg[maxn],rmq2[maxn][20];
void init( lint s,lint t ){
    lint dt = t-s+1;
    lg[1] = 0;
    for( lint i = 2;i <= dt;i++ ) lg[i] = i&(i-1) ? lg[i-1] : lg[i-1] + 1;
}
void build( const suffix& g, lint s,lint t,lint rmq[][20] ){
    lint dt = t-s;
    for( lint i = s;i < t;i++ ){
        rmq[i][0] = g.height[i];
    }
    for( lint i = 1;i <= lg[dt];i++ ){
        for( lint j = s;j + (1 << i) <= t;j++ ){
            rmq[j][i] = min( rmq[j][ i-1 ],rmq[ j + (1 << i-1  ) ][ i-1 ] );
        }
    }
}
lint lcp( const suffix& g,lint x,lint y,lint rmq[][20] ) {
    x = g.rk[x];
    y = g.rk[y];
    x++;
    y++;
    if (x > y) swap(x, y);
    lint k = lg[y - x];
    return min(rmq[x][k], rmq[y - (1 << k)][k]);
}
char str[maxn];
int st[maxn],ed[maxn];
int main(){
    int T;
    init( 0,maxn-2 );
    scanf("%d",&T);
    while(T--){
        scanf("%s",str);
        int len = strlen(str);
        memset( st,0,sizeof(int)*(len+1) );
        memset( ed,0,sizeof(int)*(len+1) );
        for( int i = 0;i < len;i++ ) g.s[i] = str[i]-'a';
        for( int i = len-1;i >= 0;i-- ) g2.s[len-1-i] = str[i]-'a';
        g.Build_SA(len);g2.Build_SA(len);
        g.getheight();g2.getheight();
        build( g,0,len,rmq );
        build( g2,0,len,rmq2 );
        for( int i = 1;i <= len/2;i++ ){
            for( int j = 0;j + i <= len-1;j+= i ){
                int su = lcp(g, j,j+i,rmq );
                int pre = lcp(g2,len-1-j,len-1-j-i,rmq2 );
                if( su + pre-1 < i ) continue;
                int s2 = min( j,j-(i-su) );
                int s1 = max( j-i+1,j-pre+1 );
                st[s1]++;st[s2+1]--;
                int t2 = min( j+2*i-1,j+i+su-1 );
                int t1 = max( j+i,j+i+(i-pre) );
                ed[t1]++;ed[t2+1]--;
            }
        }
        for( int i = 1;i < len;i++ ) st[i] += st[i-1];
        for( int i = 1;i < len;i++ ) ed[i] += ed[i-1];
        long long ans = 0;
        for( int i = 1;i < len-1;i++ ){
            ans += 1LL *ed[i]*st[i+1];
        }
        printf("%lld\n",ans);
    }
    return 0;
}