SPOJ 2798 Query on a tree again 树链剖分

这题真是神题！

当然我不是说这题有多神，而是你数组开小了和开大了得分不同，用不同版本的编译器得分不同。。。（数组没有越界的情况下）

呜呜。。。至今没有ac。一直83分。。。抑郁了。。。

跪求神犇指明错误。。。

 

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cstdio>

#define N 151000
#define M 251000
#define INF 1e9

using namespace std;

struct RT
{
    int mn,zb;
}rt[N<<2];

int head[N],to[M],next[M];
int tot,n,qu,cnt;
int fa[N],son[N],sz[N],dep[N],top[N],q[N];
int bh[N],anbh[N];

inline void add(int u,int v)
{
    to[cnt]=v; next[cnt]=head[u]; head[u]=cnt++;
}

inline void init()
{
    memset(head,-1,sizeof head); cnt=0;
    tot=0;
}

inline void prep()
{
    int h=1,t=2,sta;
    q[1]=1; dep[1]=1;
    while(h<t)
    {
        sta=q[h++]; sz[sta]=1;
        for(int i=head[sta];~i;i=next[i])
            if(to[i]!=fa[sta])
            {
                fa[to[i]]=sta;
                dep[to[i]]=dep[sta]+1;
                q[t++]=to[i];
            }
    }
    for(int j=t-1;j>=1;j--)
    {
        sta=q[j];
        for(int i=head[sta];~i;i=next[i])
            if(to[i]!=fa[sta])
            {
                sz[sta]+=sz[to[i]];
                if(sz[to[i]]>sz[son[sta]]) son[sta]=to[i]; 
            }
    }
    for(int i=1;i<t;i++)
    {
        sta=q[i];
        if(son[fa[sta]]==sta) top[sta]=top[fa[sta]];
        else top[sta]=sta;
    }
}

inline void rewrite()
{
    for(int i=1;i<=n;i++)
        if(top[i]==i)
            for(int j=i;j;j=son[j])
            {
                bh[j]=++tot;
                anbh[tot]=j;
            }
}

inline void pushup(int u)
{
    if(rt[u<<1].mn<rt[u<<1|1].mn) rt[u]=rt[u<<1];
    else rt[u]=rt[u<<1|1];
}

inline void build(int u,int L,int R)
{
    if(L==R) {rt[u].mn=INF;rt[u].zb=L;return;}
    int MID=(L+R)>>1;
    build(u<<1,L,MID); build(u<<1|1,MID+1,R);
    pushup(u);
}

inline void read()
{
    init();
    scanf("%d%d",&n,&qu);
    for(int i=1,a,b;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b); add(b,a);
    }
    prep();
    rewrite();
    build(1,1,tot);
}

inline int queryval(int u,int L,int R,int pos)
{
    if(L==R) return rt[u].mn;
    int MID=(L+R)>>1;
    if(pos<=MID) return queryval(u<<1,L,MID,pos);
    return queryval(u<<1|1,MID+1,R,pos);
}

inline void updata(int u,int L,int R,int pos,int val)
{
    if(L==R) {rt[u].mn=val;return;}
    int MID=(L+R)>>1;
    if(pos<=MID) updata(u<<1,L,MID,pos,val);
    else updata(u<<1|1,MID+1,R,pos,val);
    pushup(u);
}

inline void change(int x)
{
    int val=queryval(1,1,tot,bh[x]);
    if(val==INF) updata(1,1,tot,bh[x],dep[x]);
    else updata(1,1,tot,bh[x],INF);
}

inline RT querymin(int u,int L,int R,int l,int r)
{
    if(l<=L&&R<=r) return rt[u];
    int MID=(L+R)>>1; RT lp,rp; lp.mn=rp.mn=INF;
    if(l<=MID) lp=querymin(u<<1,L,MID,l,r);
    if(MID<r) rp=querymin(u<<1|1,MID+1,R,l,r);
    if(lp.mn<rp.mn) return lp;
    return rp;
}

inline void query(int x,int y)
{
    RT tmp,res; res.mn=INF;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        tmp=querymin(1,1,tot,bh[top[x]],bh[x]);
        if(tmp.mn<res.mn) res=tmp;
        x=fa[top[x]];
    }
    if(bh[x]>bh[y]) swap(x,y);
    tmp=querymin(1,1,tot,bh[x],bh[y]);
    if(tmp.mn<res.mn) res=tmp;
    if(res.mn!=INF) printf("%d\n",anbh[res.zb]);
    else puts("-1");
}

inline void go()
{
    int a,b;
    while(qu--)
    {
        scanf("%d%d",&a,&b);
        if(a==0) change(b);
        else query(b,1);
    }
}

int main()
{
    read(),go();
    return 0;
}