poj 3145 Harmony Forever(线段树)

poj 3145 Harmony Forever

奇葩怪异吐血题，无力多说、、、

#include<stdio.h>
#include<string.h>
#define MAXN 600000
#define INF 2000000
#define lch p<<1
#define rch p<<1|1
#define mid (t[p].l+t[p].r)>>1
struct node
{
    int l,r;
    int val;
}t[MAXN<<2];
void construct(int l,int r,int p)
{
    t[p].l=l,t[p].r=r,t[p].val=INF;
    if(l==r) return ;
    int m=mid;
    construct(l,m,lch);
    construct(m+1,r,rch);
}
int min(int a,int b) {return a<b?a:b;}
void modify(int x,int p)
{
    if(t[p].l==t[p].r)
    {
        t[p].val=x;
        return ;
    }
    int m=mid;
    if(x<=m) modify(x,lch);
    else modify(x,rch);
    t[p].val=min(t[lch].val,t[rch].val);
}
void query(int l,int r,int p,int &num)
{
    if(t[p].l>=l&&t[p].r<=r)
    {
        num=min(num,t[p].val);
        return ;
    }
    int m=mid;
    if(l<=m) query(l,r,lch,num);
    if(r>m) query(l,r,rch,num);
}
int rec[MAXN],id[MAXN];
int main()
{
    int n,x,cas=1;
    char op[3];
    while(scanf("%d",&n)!=EOF&&n)
    {
        if(cas>1) printf("\n");
        printf("Case %d:\n",cas++);
        construct(1,MAXN,1);
        int up=0,max=0;
        memset(id,-1,sizeof(id));
        while(n--)
        {
            scanf("%s%d",op,&x);
            if(op[0]=='B'&&id[x]==-1)
            {
                rec[up]=x;
                id[x]=up;
                modify(x,1);
                up++;
                if(x>max) max=x;
            }
            else
            {
                int ans=-1,re=x;
                if(x<5000)
                {
                    for(int i=up-1;i>=0;i--)
                    {
                        if(rec[i]%x<re)
                        {
                            re=rec[i]%x;
                            ans=id[rec[i]];
                        }
                        if(re==0) break;
                    }
                }
                else
                {
                    int dd=MAXN/x;
                    int re=x;
                    for(int i=0;i<=dd;i++)
                    {
                        int num=INF;
                        int l=i*x;
                        if(l<1) l=1;
                        int r=(i+1)*x-1;
                        if(r>MAXN) r=MAXN;
                        query(l,r,1,num);
                        if(num==INF) continue;
                        if(num%x<re)
                        {
                            re=num%x;
                            ans=id[num];
                        }
                        else if(num%x==re)
                        {
                            if(id[num]>ans) ans=id[num];
                        }
                    }
                }
                if(ans!=-1) ans++;
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}