HDU 5371 Hotaru's problem

HDU 5371 Hotaru's problem

比賽時這道題T了n多次，一直到比賽結束都沒有過

最後題解上說要用set優化，事實果然如此就這樣優化後過了

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<set>
#define maxn 4000005
#define ll __int64
#define INF 0xfffffff
using namespace std;
int Ma[maxn];
int Mp[maxn];
int a[maxn];
void Manacher(int a[],int len){
    int l=0;
    Ma[l++]=-2;
    Ma[l++]=-1;
    for(int i=0;i<len;i++){
        Ma[l++]=a[i];
        Ma[l++]=-1;
    }
    Ma[l]=-3;
    int mx=0,id=0;
    for(int i=0;i<l;i++){
        Mp[i]=mx>i?min(Mp[2*id-i],mx-i):1;
        while(Ma[i+Mp[i]]==Ma[i-Mp[i]]) Mp[i]++;
        if(i+Mp[i]>mx){
            mx=i+Mp[i];
            id=i;
        }
    }
}
struct node{
    int a,i;
}p[maxn];
bool cmp(node a,node b){
    if(a.a!=b.a) return a.a>b.a;
    return a.i>b.i;
}
set<int > s;
set<int > ::iterator it;
int main(){
    int T,n;
    scanf("%d",&T);
    for(int h=1;h<=T;h++){
        scanf("%d",&n);
        for(int i=0;i<n;i++) scanf("%d",&a[i]);
        Manacher(a,n);
        int ans=0,top=2*n+2,l,r;
        int A=1;
        for(int i=1;i<top;i+=2)
            Mp[A++]=Mp[i]/2;
        for(int i=1;i<=n;i++){
            p[i-1].a=Mp[i];
            p[i-1].i=i;
        }
        sort(p,p+n,cmp);
        s.clear();
        for(int i=0;i<n;i++){
            int u=p[i].a+p[i].i;
            s.insert(p[i].i);
            it=s.upper_bound(u);
            it--;
            ans=max(ans,(*it)-p[i].i);
            int v=p[i].i-p[i].a;
            it=s.lower_bound(v);
            ans=max(ans,p[i].i-(*it));
        }
        printf("Case #%d: %d\n",h,ans*3);
    }
    return 0;
}
/**
100
15
1 2 3 4 5 5 4 3 2 1 1 2 3 4 5
15
3 4 5 5 4 3 2 1 1 2 3 4 5 5 4
10
2 3 4 4 3 2 2 3 4 4
11
1 1 2 3 3 2 1 1 2 3 3

*/