Hdu 6230 树状数组+manacher

思路：满足题意必须找到一对i，j，符合下列条件

j>i

j-i<=p[i]

j-i<=p[j]

p[i]为以第i个字符为中心的回文串半径，最后一个式子一移项得j-p[j]<=i，然后通过前两个式子可以确定j的范围在(i,i+p[i]]里，每次把符合条件的j放进去，然后每次查询（i，i+p[i]]区间和就行了。

代码:

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
#define pb push_back
const int maxn=5e5+9;
int p[maxn*2],c[maxn],pp[maxn],n;
vector<int>g[maxn];
int manacher(string s){
    string s_new;
    int k=0,mx=0,id=0;
    s_new+='&';s_new+='#';
    for(int i=0;i<s.size();i++){
        s_new+=s[i];
        s_new+='#';
        k+=2;
    }
    for(int i=0;i<k;i++){
        if(i<mx)p[i]=min(p[2*id-i],mx-i);
        else p[i]=1;
        while(s_new[i+p[i]]==s_new[i-p[i]]){
            p[i]++;
        }
        if(mx<i+p[i]){
            id=i;
            mx=i+p[i];
        }
    }
    return k;
}
int lowbit(int x){
    return x&-x;
}
void update(int i,int val){
    while(i<=n){
        c[i]+=val;
        i+=lowbit(i);
    }
}
int sum(int i){
    int ans=0;
    while(i>0){
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int i,j,k,t;
    cin>>t;
    while(t--){
        string s;cin>>s;
        memset(c,0,sizeof(c));
        for(i=0;i<maxn;i++)g[i].clear();
        int k=1,len=manacher(s);
        for(i=2;i<len;i+=2){
            pp[k]=p[i]/2-1;
            g[k-pp[k]].pb(k);
            k++;
        }
        ll ans=0;
        n=s.size();
        for(i=1;i<=s.size();i++){
            for(j=0;j<g[i].size();j++)update(g[i][j],1);//仔细分析可以发现g[i]里面的元素必定满足且只满足t-pp[t]==i
            //然而对于一个i,满足t'-pp[t']==i'(i'<i)已经在之前放进去了，所以我们每次都相当于把满足条件的元素都放进去了
            ans+=sum(min(i+pp[i],n))-sum(i);
        }
        cout<<ans<<endl;
    }
}