【最长回文子串】【平衡树】[HDU5371]Hotaru's problem

题目
分析：用manacher求出以每个字符为中心的最长回文字串。
然后进行枚举。
若枚举到下标i,我们用两棵平衡树来存储在i前面，j+len[j]≥i的节点，其中一个按照j排序，另一个按照j+len[j]排序，根据第二棵平衡树的顺序进行遍历，将j+len[j]<i的节点在两棵平衡树中同时删掉，然后在第一棵平衡树中找一个最小的，更新答案。

#include<cstdio>
#include<algorithm>
#include<queue>
#include<ctime>
#include<cstring>
#define MAXN 100000
int b[MAXN*2+10],len[MAXN*2+10],ans,na,n,s[MAXN*2+10];
int r[MAXN+10],nr,T,pre;
using namespace std;
void Read(int &x){
    char c;
    while(c=getchar(),c!=EOF)
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            return;
        }
}
void read(){
    Read(n);
    na=n;
    int i;
    for(i=0;i<n;i++)
        Read(s[i]),r[i]=s[i];
    sort(r,r+n);
    nr=unique(r,r+n)-r;
    for(i=n-1;i>=0;i--){
        s[i]=lower_bound(r,r+nr,s[i])-r+1;
        s[i*2+2]=s[i],s[i*2+1]=-1;
    }
    s[n*2+2]=-2,s[0]=-3,s[n*2+1]=-1;
    na=n;
    n=(n+1)<<1;
}
struct node{
    int val,pri;
    node *ch[2];
}tree[MAXN*2+10],*tcnt=tree,*root1,*root2;
void Rotate(node *&x,bool d){
    node *y=x->ch[!d];
    x->ch[!d]=y->ch[d];
    y->ch[d]=x;
    x=y;
}
void insert1(node *&p,int val){
    if(!p){
        p=++tcnt;
        p->val=val;
        p->pri=rand();
        p->ch[0]=p->ch[1]=0;
        return;
    }
    bool d=val>p->val;
    insert1(p->ch[d],val);
    if(p->ch[d]->pri<p->pri)
        Rotate(p,!d);
}
void insert2(node *&p,int val){
    if(!p){
        p=++tcnt;
        p->val=val;
        p->pri=rand();
        p->ch[0]=p->ch[1]=0;
        return;
    }
    bool d=val+len[val]>=len[p->val]+p->val;
    insert2(p->ch[d],val);
    if(p->ch[d]->pri<p->pri)
        Rotate(p,!d);
}
void del(node *&p,int val){
    if(p->val==val){
        if(!p->ch[0])
            p=p->ch[1];
        else if(!p->ch[1])
            p=p->ch[0];
        else if(p->ch[0]->pri<p->ch[1]->pri){
            Rotate(p,1);
            del(p->ch[1],val);
        }
        else{
            Rotate(p,0);
            del(p->ch[0],val);
        }
        return;
    }
    if(val>p->val)
        del(p->ch[1],val);
    else
        del(p->ch[0],val);
}
void find_pre(node *p,int val){
    if(!p)
        return;
    if(val>p->val)
        find_pre(p->ch[1],val);
    else if(val<p->val){
        pre=min(pre,p->val);
        find_pre(p->ch[0],val);
    }
    else
        pre=val;
}
void bianli(node *&p,int i){
    if(!p)
        return;
    bianli(p->ch[0],i);
    if(p->val+len[p->val]<i){
        bianli(p->ch[1],i);
        del(root1,p->val);
        del(p,p->val);
    }
}
void solve(){
    int i;
    for(i=1;i<na;i++){
        bianli(root2,i);
        if(root1){
            pre=0x7fffffff;
            find_pre(root1,i-len[i]);
            if(pre!=0x7fffffff)
                ans=max(ans,(i-pre)*3);
        }
        insert1(root1,i);
        insert2(root2,i);
    }
}
void manacher(){
    int i,j,po=0,p=0;
    for(i=1;i<n;i++){
        if(p>=i){
            j=po*2-i;
            if(i+len[j]<=p){
                len[i]=len[j];
                continue;
            }
            else
                len[i]=p-i+1;
        }
        else
            len[i]=1;
        while(s[i+len[i]]==s[i-len[i]])
            len[i]++;
        p=i+len[i]-1,po=i;
    }
    for(i=3;i<n;i+=2)
        len[i/2]=len[i]>>1;
}
int main()
{
    srand(time(0));
    Read(T);
    int cnt=0;
    while(T--){
        root1=root2=0,tcnt=tree;
        ans=0;
        read();
        manacher();
        solve();
        printf("Case #%d: %d\n",++cnt,ans);
    }
}

转载于:https://www.cnblogs.com/outerform/p/5921893.html