【CCPC2017哈尔滨A】Palindrome

原题：

 

 

题中那个奇怪的式子其实就表示一个三段对称的回文串

那用马拉车求出所有回文串及半径，问题就转化为有多少个回文串的中点i和j，使得|i-j|<=min{r[i],r[j]}

可以按照r递减的顺序枚举，这样只需考虑当前枚举到的回文串内有多少个已枚举过的回文串

 

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
struct nds{int x,y;}b[510000];
int n;  char s[510000];
char ss[1100000];
int f[1100000];
int ar[510000];
int v[510000];
inline int lbt(int x){  return x&-x;}
int qry(int x){
    int bwl=0;
    while(x)  bwl+=v[x],x-=lbt(x);
    return bwl;
}
void mdf(int x){while(x<=n)  v[x]++,x+=lbt(x);}
void mlc(){
    for(int i=0;i<n;++i)  ss[i<<1]='#',ss[i<<1|1]=s[i];
    int nn=n<<1;  ss[nn]='#';
    int mx=0,id=0;
    for(int i=0;i<=nn;++i){
        f[i]=(mx>i) ? min(f[2*id-i],mx-i) : 1;
        while(i>=f[i] && i+f[i]<=nn && ss[i+f[i]]==ss[i-f[i]])  ++f[i];
        if(mx<i+f[i]-1)  mx=i+f[i]-1,id=i;
    }
    for(int i=0;i<n;++i)  ar[i]=f[i<<1|1]/2-1;
    //for(int i=0;i<n;++i)  cout<<ar[i]<<" ";
    //cout<<endl;
}
bool cmp(nds x,nds y){
    return x.x==y.x ? x.y<y.y : x.x<y.x;
}
void prvs(){
    for(int i=0;i<=n;++i)  v[i]=0;
}
int main(){
    int T;  cin>>T;
    while(T --> 0){
        scanf("%s",s);  n=strlen(s);
        prvs();
        mlc();
        for(int i=0;i<n;++i){
            //b[i]=(nds){ar[i],i+1};
            b[i].x=ar[i];
            b[i].y=i+1;
        }
        sort(b,b+n,cmp);
        long long cnt=0;
        for(int i=n-1;i>=0;--i){
            cnt+=qry(min(n,b[i].y+b[i].x))-qry(max(1,b[i].y-b[i].x)-1);
            mdf(b[i].y);
        }
        printf("%lld\n",cnt);
    }
    return 0;
}