BZOJ1146 [CTSC2008]网络管理Network

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

int in(){
    int x=0;char ch=getchar();
    while(ch>‘9‘ || ch<‘0‘) ch=getchar();
    while(ch>=‘0‘ && ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();
    return x;
}

const int maxn=80010;

struct Spot{
    int next,data;
}spot[maxn<<1];

#define now spot[pt].data
#define then spot[pt].next

struct Node{
    int l,r,sz;
}s[maxn*120];

int n,m,cnt,key,Sz,ans;
int a[maxn],K[maxn],A[maxn],B[maxn];
int num[maxn<<1],tp,Hash[maxn<<1],hp;
int order[maxn],tail,dic[maxn],nxt[maxn];
int head[maxn],rt[maxn];
int f[maxn][17],depth[maxn];
int Q[4][17];

void add(int u,int v){
    spot[cnt].data=v;spot[cnt].next=head[u];head[u]=cnt++;
    spot[cnt].data=u;spot[cnt].next=head[v];head[v]=cnt++;
}

inline int get_h(int x){
    int l=0,r=hp+1,mid;
    while(l<r-1){
        mid=(l+r)>>1;
        if(Hash[mid]>x) r=mid;
        else l=mid;
    }
    return l;
}

void update(int l,int r,int &rt,int val){
    if(!rt) rt=++Sz; s[rt].sz+=val;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(key<=mid) update(l,mid,s[rt].l,val);
    else update(mid+1,r,s[rt].r,val);
}

inline void renew(int x,int val){
    for(int i=x;i<=n;i+=i&(-i))
        update(1,hp,rt[i],val);
}

void dfs(int last,int x){
    order[++tail]=x;dic[x]=tail;
    for(int i=1;i<17;i++)
        f[x][i]=f[f[x][i-1]][i-1];
    for(int pt=head[x];pt!=-1;pt=then)
        if(now!=last)
            f[now][0]=x,depth[now]=depth[x]+1,dfs(x,now);
    nxt[x]=tail+1;
}

int LCA(int x,int y){
    if(depth[x]<depth[y]) swap(x,y);
    int t=depth[x]-depth[y];
    for(int i=16;i>=0;i--)
        if(t&(1<<i)) x=f[x][i];
    if(x==y) return x;
    for(int i=16;i>=0;i--)
        if(f[x][i]!=f[y][i])
            x=f[x][i],y=f[y][i];
    return f[x][0];
}

int ask(int l,int r,int k){
    if(l==r) return l;
    int sum=0,mid=(l+r)>>1;
    for(int i=0;i<2;i++)
    for(int j=1;j<=Q[i][0];j++) sum+=s[s[Q[i][j]].l].sz;
    for(int i=2;i<4;i++)
    for(int j=1;j<=Q[i][0];j++) sum-=s[s[Q[i][j]].l].sz;
    if(k<=sum){
        for(int i=0;i<4;i++)
        for(int j=1;j<=Q[i][0];j++) Q[i][j]=s[Q[i][j]].l;
        return ask(l,mid,k);
    }
    else{
        for(int i=0;i<4;i++)
        for(int j=1;j<=Q[i][0];j++) Q[i][j]=s[Q[i][j]].r;
        return ask(mid+1,r,k-sum);
    }
}

int query(int k,int l,int r){
    int t=LCA(l,r),All=depth[l]+depth[r]-(depth[t]<<1)+1;
    if(All<k) return -1;
    Q[0][0]=Q[1][0]=Q[2][0]=Q[3][0]=0;
    for(int i=dic[l];i;i-=i&-i) Q[0][++Q[0][0]]=rt[i];
    for(int i=dic[r];i;i-=i&-i) Q[1][++Q[1][0]]=rt[i];
    for(int i=dic[t];i;i-=i&-i) Q[2][++Q[2][0]]=rt[i];
    for(int i=dic[f[t][0]];i;i-=i&-i) Q[3][++Q[3][0]]=rt[i];
    return Hash[ask(1,hp,All-k+1)];
}

void modify(int x,int y){
    key=a[x],renew(dic[x],-1),renew(nxt[x],1);
    key=a[x]=get_h(y),renew(dic[x],1),renew(nxt[x],-1);
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("114610.in","r",stdin);
    freopen("1146.out","w",stdout);
#endif
    
    int u,v,p;
    
    n=in();m=in();
    for(int i=1;i<=n;i++)
        a[i]=in(),num[++tp]=a[i],head[i]=-1;
    for(int i=1;i<n;i++)
        u=in(),v=in(),add(u,v);
    for(int i=1;i<=m;i++){
        K[i]=in(),A[i]=in(),B[i]=in();
        if(!K[i]) num[++tp]=B[i];
    }
    sort(num+1,num+tp+1);
    Hash[++hp]=num[1];
    for(int i=2;i<=tp;i++)
        if(num[i]!=num[i-1]) Hash[++hp]=num[i];
    
    dfs(0,1);
    for(int i=1;i<=n;i++) a[i]=get_h(a[i]);
    for(int i=1;i<=n;i++)
        key=a[i],renew(dic[i],1),renew(nxt[i],-1);
    
    for(int i=1;i<=m;i++)
        if(K[i]){
            ans=query(K[i],A[i],B[i]);
            if(ans>0) printf("%d\n",ans);
            else puts("invalid request!");
        }
        else modify(A[i],B[i]);

    return 0;
}