HDU-6230 - Palindrome （Manacher + 主席树/树状数组）

Palindrome

题目链接：Palindrome HDU - 6230

题意

给你一个字符串，问你里面会有几个符合一个半回文串的定义，一个半回文串的定义是类似与这种abcbabc ；

---

思路

• 题意可以转化为:有多少对i,j(i < j)满足S[i-(j-i)…j],S[i…j+(j-i)]为回文串
• 即以i，j为中心的回文串能够相互覆盖到
• 我们设len[i]表示，以i为中心的回文半径可以用马拉车求解
• 相互覆盖即：
• i+len[i]>=j
• J-len[j]<=I

由于i < j，然后就可以转化为区间求有多少个数小于x的问题，n很大大概不能主席树，树状数组离线即可。

可惜我那时对于树状数组有点忘了，所以我用主席树硬刚了一发，hhA了。

这里给出三种实现代码

1. 主席树就是求区间有多少数大于等于x，不过有个陷阱，题目求得是有多少组，如果两个x相当，我们会重复计算，我这里就用来一个技巧，在一个区间内左半边找大于等于x的数，而右半边找大于x的数。这样就不会重复计算了。
2. 自己后来想的树状数组，我们可以先将N个询问离线，先询问覆盖面积大的，在询问小的，这样就不会重复了。
3. 看大佬的树状数组，将询问的left从小到大排序，每次访问 i 到 i这点的r，每次将left用while调整至跟i同一大小。

---

代码一

#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)(j);i <= (int)(k);i ++)
#define per(i,j,k) for(int i = (int)(j);i >= (int)(k);i --)
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define mmm(a,b)   memset(a,b,sizeof(a))
#define pb push_back

typedef long long ll;
const int INF = (int)0x3f3f3f3f;
const int MaxLen = (int)1e6+1007;

char ptr[MaxLen];
char ntr[MaxLen];
int p[MaxLen];
int len1,len2;

void MakeNtr(){
    len2 = len1*2+2;
    ntr[0] = '$';
    ntr[1] = '#';
    rep(i,0,len1-1){
        ntr[(i+1)*2] = ptr[i];
        ntr[(i+1)*2+1] =  '#';
    }
    ntr[len2] = '\0';
}

void Manacher(){
    MakeNtr();
    int res = 0;
    int id,mx = 0;
    rep(i,0,len2-1){
        if (i < mx) p[i] = min(p[id*2-i],mx-i);
        else        p[i] = 1;
        while (ntr[i+p[i]] == ntr[i-p[i]])p[i] ++;
        if (mx < i+p[i]){
            mx = i+p[i];
            id = i;
        }
        p[i]-1;
    }
}


const int MAXN = (int)5e5+7;

int N,M;
int A[MAXN];
int tree[MAXN*40],lc[MAXN*40],rc[MAXN*40],root[MAXN*40],tot;
vector<int> vp;
int getID(int x) {return upper_bound(vp.begin(),vp.end(),x)-vp.begin(); }

void Update(int last,int cur,int l,int r,int pos) {
    tree[cur] = tree[last];
    lc[cur] = lc[last];
    rc[cur] = rc[last];
    if (l == r) {
        tree[cur] ++;
        return ;
    }
    int m = l+r>>1;
    if (pos <= m) Update(lc[last],lc[cur]=++tot,l,m,pos);
    else          Update(rc[last],rc[cur]=++tot,m+1,r,pos);
    tree[cur] = tree[lc[cur]] + tree[rc[cur]];
}

int Query(int last,int cur,int l,int r,int k) {
    if (l == r)  {
        return tree[cur] - tree[last];
    }
    int m = l+r>>1;
    int res = 0;
    if (k <= m) res += Query(lc[last],lc[cur],l,m,k)+ tree[rc[cur]] - tree[rc[last]];
    else        res += Query(rc[last],rc[cur],m+1,r,k) ;
    return res;
}

int debugt(int cur,int l,int r) {
    if (l == r) return 0*printf("%d:%d ",l,tree[cur]);
    int m = l+r>>1;
    debugt(lc[cur],l,m);
    debugt(rc[cur],m+1,r);
}

void init() {
    vp.clear();
    tot = 0;
    tree[0] = 0;
}

int main()
{
    int T;
    scanf("%d",&T);
    rep(ca,1,T) {
        init();
        scanf("%s",ptr);
        len1 = strlen(ptr);
        Manacher();
        N = (len2-1)/2;
        rep(i,1,N) A[i] = (p[i*2]-1)/2,vp.pb(A[i]);
        sort(vp.begin(),vp.end());vp.erase(unique(vp.begin(),vp.end()),vp.end());
        rep(i,1,N) Update(root[i-1],root[i]=++tot,1,N,getID(A[i]));

        ll res = 0;
        rep(i,1,N) {
            int l,r,k;
            l = i-A[i];
            r = i+A[i];
            k = getID(A[i]);
            if (A[i] == 0) continue;

            res += (ll)Query(root[i],root[r],1,N,k+1);
            res += (ll)Query(root[l-1],root[i-1],1,N,k);
        }
        printf("%lld\n",res);
    }
}

代码二

#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)(j);i <= (int)(k);i ++)
#define per(i,j,k) for(int i = (int)(j);i >= (int)(k);i --)
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define mmm(a,b)   memset(a,b,sizeof(a))
#define pb push_back

typedef long long ll;
const int INF = (int)0x3f3f3f3f;
const int MaxLen = (int)1e6+1007;

char ptr[MaxLen];
char ntr[MaxLen];
int p[MaxLen];
int len1,len2;

inline void MakeNtr(){
    len2 = len1*2+2;
    ntr[0] = '$';
    ntr[1] = '#';
    rep(i,0,len1-1){
        ntr[(i+1)*2] = ptr[i];
        ntr[(i+1)*2+1] =  '#';
    }
    ntr[len2] = '\0';
}

inline void Manacher(){
    MakeNtr();
    int res = 0;
    int id,mx = 0;
    rep(i,0,len2-1){
        if (i < mx) p[i] = min(p[id*2-i],mx-i);
        else        p[i] = 1;
        while (ntr[i+p[i]] == ntr[i-p[i]])p[i] ++;
        if (mx < i+p[i]){
            mx = i+p[i];
            id = i;
        }
        p[i]-1;
    }
}
const int MAXN = (int) 1e6;
int N,Tr[MAXN];

struct Node{
    int l,r,len,id;
    Node(int l = 0,int r = 0,int len = 0,int id = 0):l(l),r(r),len(len),id(id){}
}Que[MAXN];
bool cmp(const Node&a,const Node&b) {
    return a.len > b.len;
}

int lowbit(int x) {return x&-x ;}

void Update(int L,int x) {
    while (L <= N) {
        Tr[L] += x;
        L += lowbit(L);
    }
}

int query(int L) {
    int sum = 0;
    while (L > 0) {
        sum += Tr[L];
        L -= lowbit(L);
    }
    return sum;
}

void init() {
    mmm(Tr,0);
}

int main()
{
    int T;
    scanf("%d",&T);
    rep(ca,1,T) {
        init();
        scanf("%s",ptr);
        len1 = strlen(ptr);
        Manacher();
        N = (len2-1)/2;
        rep(i,1,N) {
            int len = (p[i*2]-1)/2;
            int l = i-len;
            int r = i+len;
            //debug(l),debug(r),debug(len);
            Que[i] = Node(l,r,len,i);
        }
        sort(Que+1,Que+1+N,cmp);
        ll res = 0;
        rep(i,1,N) {
            res += (ll)(query(Que[i].r) - query(Que[i].l-1));
            Update(Que[i].id,1);
        }
        printf("%lld\n",res);
    }
}

代码三

#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)(j);i <= (int)(k);i ++)
#define per(i,j,k) for(int i = (int)(j);i >= (int)(k);i --)
#define debug(x) cerr<<#x<<" = "<<(x)<<endl
#define mmm(a,b)   memset(a,b,sizeof(a))
#define pb push_back

typedef long long ll;
const int INF = (int)0x3f3f3f3f;
const int MaxLen = (int)1e6+1007;

char ptr[MaxLen];
char ntr[MaxLen];
int p[MaxLen];
int len1,len2;

void MakeNtr(){
    len2 = len1*2+2;
    ntr[0] = '$';
    ntr[1] = '#';
    rep(i,0,len1-1){
        ntr[(i+1)*2] = ptr[i];
        ntr[(i+1)*2+1] =  '#';
    }
    ntr[len2] = '\0';
}

void Manacher(){
    MakeNtr();
    int res = 0;
    int id,mx = 0;
    rep(i,0,len2-1){
        if (i < mx) p[i] = min(p[id*2-i],mx-i);
        else        p[i] = 1;
        while (ntr[i+p[i]] == ntr[i-p[i]])p[i] ++;
        if (mx < i+p[i]){
            mx = i+p[i];
            id = i;
        }
    }
}
const int MAXN = (int) 1e6;
int N,Tr[MAXN];

struct Node{
    int l,r,len,id;
    Node(int l = 0,int r = 0,int id = 0):l(l),r(r),id(id){}
}Que[MAXN];
bool cmp(const Node&a,const Node&b) {
    return a.l < b.l;
}

int lowbit(int x) {return x&-x ;}

void Update(int L,int x) {
    while (L <= N) {
        Tr[L] += x;
        L += lowbit(L);
    }
}

int query(int L) {
    int sum = 0;
    while (L > 0) {
        sum += Tr[L];
        L -= lowbit(L);
    }
    return sum;
}

void init() {
    mmm(Tr,0);
}

int main()
{
    int T;
    scanf("%d",&T);
    rep(ca,1,T) {
        init();
        scanf("%s",ptr);
        len1 = strlen(ptr);
        Manacher();
        N = (len2-1)/2;
        rep(i,1,N) {
            int len = (p[i*2]-1)/2;
            int l = i-len;
            int r = i+len;
            //debug(l),debug(r),debug(len);
            Que[i] = Node(l,r,i);
        }
        sort(Que+1,Que+1+N,cmp);
        ll res = 0;
        int j = 1;
        rep(i,1,N) {
            //debug(Que[i].l),debug(Que[i].r),debug(Que[i].id);cout << endl;
            while (Que[j].l <= i && j <= N) Update(Que[j].id,1),j++;
            res += (ll) (query((p[i*2]-1)/2+i) - query(i));
        }
        printf("%lld\n",res);
    }
}