bzoj1146 [CTSC2008]网络管理（树上主席树，树状数组套动态加点线段树）

给定一棵树，查询路径第k大值，带修改。
就是bzoj2588+bzoj3196
树上建主席树。然后对于修改，修改点x的权值，只会对x的子树有影响，而x的子树在dfs序上是连续的一段in[x],out[x]，就相当于区间修改，我们利用差分+树状数组来维护，另建bit+动态加点线段树来维护。
树状数组套动态加点线段树内存究竟怎么算.jpg???能开多大是多大好了，反正是要比logn^2小很多？这样写好写还跑得快，就是内存比较迷

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 80010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,a[N],aa[N<<1],num=0,h[N],tot=0,owo=0,rt[N],brt[N];
int fa[N][18],in[N],out[N],dfn=0,dep[N],Log[N],p1[N],p2[N],n1,n2;
struct edge{
    int to,next;
}data[N<<1];
struct event{
    int k,a,b;
}e[N];
struct node{
    int lc,rc,sum;
}tr[N*128];
inline void ins(int &p,int l,int r,int x){
    tr[++owo]=tr[p];p=owo;tr[p].sum++;
    if(l==r) return;int mid=l+r>>1;
    if(x<=mid) ins(tr[p].lc,l,mid,x);
    else ins(tr[p].rc,mid+1,r,x);
}
inline void dfs(int x){
    in[x]=++dfn;rt[x]=rt[fa[x][0]];ins(rt[x],1,tot,a[x]);
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x][0]) continue;
        fa[y][0]=x;dep[y]=dep[x]+1;dfs(y);
    }out[x]=dfn;
}
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=0;i<=Log[d];++i)
        if(d>>i&1) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
inline void add(int &p,int l,int r,int x,int val){
    if(!p) p=++owo;tr[p].sum+=val;
    if(l==r) return;int mid=l+r>>1;
    if(x<=mid) add(tr[p].lc,l,mid,x,val);
    else add(tr[p].rc,mid+1,r,x,val);
}
inline void bit_add(int x,int pos,int val){
    for(;x<=n;x+=x&-x) add(brt[x],1,tot,pos,val);
}
inline void gao1(int x){
    for(;x;x-=x&-x) p1[++n1]=brt[x];
}
inline void gao2(int x){
    for(;x;x-=x&-x) p2[++n2]=brt[x];
}
inline int qkth(int l,int r,int k){
    if(l==r) return l;int mid=l+r>>1,res=0;
    for(int i=1;i<=n1;++i) res+=tr[tr[p1[i]].lc].sum;
    for(int i=1;i<=n2;++i) res-=tr[tr[p2[i]].lc].sum;
    if(k<=res){
        for(int i=1;i<=n1;++i) p1[i]=tr[p1[i]].lc;
        for(int i=1;i<=n2;++i) p2[i]=tr[p2[i]].lc;
        return qkth(l,mid,k);
    }
    for(int i=1;i<=n1;++i) p1[i]=tr[p1[i]].rc;
    for(int i=1;i<=n2;++i) p2[i]=tr[p2[i]].rc;
    return qkth(mid+1,r,k-res);
}
inline void query(int x,int y,int k){
    int t=lca(x,y);n1=n2=0;
    p1[++n1]=rt[x];p1[++n1]=rt[y];
    p2[++n2]=rt[t];p2[++n2]=rt[fa[t][0]];
    gao1(in[x]);gao1(in[y]);gao2(in[t]);gao2(in[fa[t][0]]);
    int sz=0;
    for(int i=1;i<=n1;++i) sz+=tr[p1[i]].sum;
    for(int i=1;i<=n2;++i) sz-=tr[p2[i]].sum;
    if(k>sz){puts("invalid request!");return;}
    printf("%d\n",aa[qkth(1,tot,sz-k+1)]);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();tot=n;Log[0]=-1;
    for(int i=1;i<=n;++i) a[i]=aa[i]=read(),Log[i]=Log[i>>1]+1;
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }for(int i=1;i<=m;++i){
        e[i].k=read();e[i].a=read();e[i].b=read();
        if(!e[i].k) aa[++tot]=e[i].b;
    }sort(aa+1,aa+tot+1);tot=unique(aa+1,aa+tot+1)-aa-1;
    for(int i=1;i<=n;++i) a[i]=lower_bound(aa+1,aa+tot+1,a[i])-aa;
    dfs(1);for(int i=1;i<=m;++i){
        if(!e[i].k){
            int x=e[i].a,val=lower_bound(aa+1,aa+tot+1,e[i].b)-aa;
            bit_add(in[x],a[x],-1);bit_add(out[x]+1,a[x],1);a[x]=val;
            bit_add(in[x],a[x],1);bit_add(out[x]+1,a[x],-1);
        }else query(e[i].a,e[i].b,e[i].k);
    }return 0;
}