2017 ccpc哈尔滨 A题 Palindrome

2017 ccpc哈尔滨 A题 Palindrome

题意：

给一个串\(T\),计算存在多少子串S满足\(S[i]=S[2n−i]=S[2n+i−2](1≤i≤n)\)

思路：

很明显这里的回文串长度为奇数，所以用\(manacher\)处理时不需要添加间隔字符

所以这里的\(Len[i]\)表示的就是以\(i\)为中心的回文串向左右最远能延伸的长度

那么\(S[i]=S[2n−i]=S[2n+i−2](1≤i≤n)\)就等价于

找到一对$(i,j), 满足i - Len[i] + 1 <= j < i 且 j + Len[j] - 1 >= i \(
可以用主席树来维护，更简单的方法就是
将\)j + Len[j] - 1按升序排列\(,然后对于\)j$丢到树状数组里查询贡献就好了。

#include<bits/stdc++.h>
#define P pair<int,int>
#define LL long long
using namespace std;
const int maxn = 5e5 + 10;
const int N = 1e6 + 10;
char s[N];
int Len[N];
int lowbit(int x){return x & (-x);}
int tr[N],R;
int getsum(int pos){
    int ans = 0;
    for(;pos;pos -= lowbit(pos)) ans += tr[pos];
    return ans;
}
void up(int pos){
    for(;pos <= R;pos += lowbit(pos)) tr[pos]++;
}
void Manacher(char *s){
    int len = strlen(s + 1);
    s[0] = '#';
    int mx = 0,center = 0;
    ///mx为当前计算回文串最右边字符的最大值
    ///center为取得mx最大值的中心
    for(int i = 1;i <= len;i++){
        if(mx > i) Len[i] = min(mx - i, Len[2 * center - i]);///考虑i关于center的对称的Len
        else Len[i] = 1;
        while(s[i - Len[i]] == s[i + Len[i]]) Len[i]++;
        if(Len[i] + i > mx) mx = Len[i] + i, center = i; ///更新最右
    }
}
struct node{
    int x,l;
    node(int x,int l):x(x),l(l){};
    node(){};
    bool operator<(const node &rhs)const{
        return l > rhs.l;
    }
}q[N];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
       scanf("%s",s + 1);
       Manacher(s);
       int len = strlen(s + 1);
       R = len;
       for(int i = 1;i <= R;i++) tr[i] = 0;
       for(int i = 1;i <= len;i++) q[i] = node(i, i + Len[i] - 1);
       sort(q + 1, q + len + 1);
       int l = 1;
       LL ans = 0;
       for(int i = len;i >= 1;i--){
            while(l <= len && q[l].l >= i) up(q[l++].x);
            ans += getsum(i - 1) - getsum(i - Len[i]);
       }
       printf("%lld\n",ans);
    }
    return 0;
}