bzoj1146: [CTSC2008]网络管理Network

传送门
树剖+线段树套平衡树
看到如此长的时限，我就放心大胆的开始(开车)了。
打代码真爽
具体实现：
1.二分答案，
2.在被轻重链剖分的树上跑线段树，
3.在每一个节点上维护平衡树用来查询区间排名。
复杂度分析：
1.二分一只log
2.树剖一只log
3.线段树一只log
4.平衡树一只log
然后就实现了Nlog^4N的做法。
四只log，四只log，跑得快，跑得快。
其实和N^2是没有任何区别的。
代码贼长。

#include<cmath>#include<cstdio>#include<cstring>#include<cstdlib>#include<iostream>#include<algorithm>#define N 100005#define M 5000005using namespace std;inline int read(){    int x=0,f=1;    char ch=getchar();    for (;ch<48||ch>57;ch=getchar()) if (ch=='-') f=-1;    for (;ch>47&&ch<58;ch=getchar()) x=x*10-48+ch;    return x;}int n,q,cnt,tot,sz,tmp,x,y;int bin[20],T[N],hash[N],f[N],a[N],b[N];int fa[N][17],son[N],dep[N],top[N],rt[N*3];int w[M],v[M],s[M],rnd[M],ls[M],rs[M];int vis[N],head[N],pl[N];struct edge{int to,next;}e[N*2];void add(int x,int y){    e[++tot]=(edge){y,head[x]};    head[x]=tot;    e[++tot]=(edge){x,head[y]};    head[y]=tot;}int find(int x){    int l=1,r=tot;    while (l<=r){        int mid=(l+r)/2;        if (hash[mid]==x) return mid;        if (hash[mid]<x) l=mid+1; else r=mid-1;    }    return l;}void dfs1(int x){    son[x]=vis[x]=1;    for (int i=1;bin[i]<=dep[x]&&i<=16;i++)        fa[x][i]=fa[fa[x][i-1]][i-1];    for (int i=head[x];i;i=e[i].next)        if (!vis[e[i].to]){            dep[e[i].to]=dep[x]+1;            fa[e[i].to][0]=x;            dfs1(e[i].to);            son[x]+=son[e[i].to];        }}void dfs2(int x,int tp){    pl[x]=++cnt; top[x]=tp;    int k=0;    for (int i=head[x];i;i=e[i].next)        if (dep[e[i].to]>dep[x]&&son[e[i].to]>son[k])            k=e[i].to;    if (k) dfs2(k,tp);    for (int i=head[x];i;i=e[i].next)        if (dep[e[i].to]>dep[x]&&e[i].to!=k)            dfs2(e[i].to,e[i].to);}int lca(int x,int y){    if (dep[x]<dep[y]) swap(x,y);    int tmp=dep[x]-dep[y];    for (int i=0;i<=16;i++)        if (bin[i]&tmp) x=fa[x][i];    for (int i=16;i>=0;i--)        if (fa[x][i]!=fa[y][i])            x=fa[x][i],y=fa[y][i];    return (x==y)?x:fa[x][0]; }void update(int k){    s[k]=s[ls[k]]+s[rs[k]]+w[k];}void rturn(int &k){    int t=ls[k]; ls[k]=rs[t]; rs[t]=k;    update(k); update(t); k=t;}void lturn(int &k){    int t=rs[k]; rs[k]=ls[t]; ls[t]=k;    update(k); update(t); k=t;}void insert(int &k,int num){    if (!k){        k=++sz; rnd[k]=rand();        w[k]=s[k]=1; v[k]=num;        return;    }    s[k]++;    if (num==v[k]){        w[k]++; return;    }    if (num<v[k]){        insert(ls[k],num);        if (rnd[ls[k]]<rnd[k]) rturn(k);    }    else{        insert(rs[k],num);        if (rnd[rs[k]]<rnd[k]) lturn(k);    }}void del(int &k,int num){    if (!k) return;    if (num==v[k]){        if (w[k]>1){            w[k]--; s[k]--;            return;        }        if (ls[k]*rs[k]==0) k=ls[k]+rs[k];        else if (rnd[ls[k]]<rnd[rs[k]]){            rturn(k); del(k,num);        }        else{            lturn(k); del(k,num);        }    }    else if (num<v[k]){        del(ls[k],num); s[k]--;    }    else{        del(rs[k],num); s[k]--;    }}void askrk(int k,int num){    if (!k) return;    if (num==v[k]){tmp+=s[rs[k]]; return;}    if (num<v[k]){        tmp+=s[rs[k]]+w[k];        askrk(ls[k],num);    }    else askrk(rs[k],num);}void change(int k,int l,int r,int pos,int x,int y){    del(rt[k],x); insert(rt[k],y);    if (l==r) return;    int mid=(l+r)/2;    if (pos<=mid) change(k*2,l,mid,pos,x,y);    else change(k*2+1,mid+1,r,pos,x,y);}void ask(int k,int l,int r,int x,int y,int num){    if (x==l&&y==r){askrk(rt[k],num); return;}    int mid=(l+r)/2;    if (y<=mid) ask(k*2,l,mid,x,y,num);    else if (x>mid) ask(k*2+1,mid+1,r,x,y,num);    else ask(k*2,l,mid,x,mid,num),ask(k*2+1,mid+1,r,mid+1,y,num);}void getrk(int x,int f,int num){    while (top[x]!=top[f]){        ask(1,1,n,pl[top[x]],pl[x],num);        x=fa[top[x]][0];    }    ask(1,1,n,pl[f],pl[x],num);}void solve(int x,int y,int rk){    int t=lca(x,y),ans=-1;    tmp=0; getrk(y,t,0); getrk(x,t,0);    if (tmp-1<rk){        printf("invalid request!\n");        return;    }    int l=1,r=tot;    while (l<=r){        int mid=(l+r)/2;        tmp=0;        getrk(x,t,mid);        getrk(y,t,mid);        if (T[t]>mid) tmp--;        if (tmp<rk) r=mid-1,ans=mid;        else l=mid+1;    }    printf("%d\n",hash[ans]);}int main(){    bin[0]=1;    for (int i=1;i<=16;i++) bin[i]=bin[i-1]*2;    n=read(); q=read();    for (int i=1;i<=n;i++)        hash[i]=T[i]=read();    for (int i=1;i<n;i++){        x=read(); y=read();        add(x,y);    }    tot=n;    dfs1(1); dfs2(1,1);    for (int i=1;i<=q;i++){        scanf("%d%d%d",&f[i],&a[i],&b[i]);        if (!f[i]) hash[++tot]=b[i];    }    sort(hash+1,hash+tot+1);    int top=1;    for (int i=2;i<=tot;i++)        if (hash[i]!=hash[i-1]) hash[++top]=hash[i];    tot=top;    for (int i=1;i<=n;i++) T[i]=find(T[i]);    for (int i=1;i<=q;i++) if (!f[i]) b[i]=find(b[i]);    for (int i=1;i<=n;i++) change(1,1,n,pl[i],0,T[i]);    for (int i=1;i<=q;i++)        if (!f[i]){            change(1,1,n,pl[a[i]],T[a[i]],b[i]);            T[a[i]]=b[i];        }        else solve(a[i],b[i],f[i]);}