bzoj 1162 network

树上的区间第k小数，以前写的主席树是一直MLE的，后来看到一种在初始化的时候的优化：直接DFS这颗树，得到每个点的主席树，然后更新的时候另外对DFS序建主席树，答案加上初始每个点的主席树，这样在初始化每个点的时候就不需要用BIT维护了，省去的空间正好AC了这个题。

呆马：

 #include <iostream>
 #include <cstdio>
 #include <algorithm>
 #include <cmath>
 #include <cstring>
 #include <vector>
 #define maxn 80008
 using namespace std;
 struct node
 {
     node *lc,*rc;
     int s;
     node () {s=;}
 }*rot[maxn],*sum[maxn],*p;
 vector <node*> st[];
 struct et
 {
     int s,t,next;
 }e[maxn*];
 int ll[maxn],rr[maxn],fir[maxn],key[maxn];
 int w[maxn*],dep[maxn],fa[maxn][];
 int x[maxn],y[maxn],z[maxn];
 int n,m,tot,low,cnt,num;
 int v[];

 node *build(int l,int r)
 {
     node *now=new node();
     if (l==r) return now;
     int mid=(l+r)>>;
     now->lc=build(l,mid);
     now->rc=build(mid+,r);
     return now;
 }

 node *change(int l,int r,int x,int z)
 {
     node *now=new node();
     now->lc=p->lc; now->rc=p->rc;
     now->s=p->s+z;
     if (l==r) return now;
     int mid=(l+r)>>;
     if (x<=mid) p=p->lc,now->lc=change(l,mid,x,z);
     else p=p->rc,now->rc=change(mid+,r,x,z);
     return now;
 }

 void adj(int x,int y,int z)
 {
     if (x==) return ;
     for (int i=x;i<=n;i+=(i&-i)) p=rot[i],rot[i]=change(,num,y,z);
 }       

 void fill(int k)
 {
     st[k].clear();
     st[k].push_back(sum[v[k]]);
     for (int i=ll[v[k]];i>;i-=(i&-i)) st[k].push_back(rot[i]);
 }

 int getsum(int k)
 {
     int ans=, n=st[k].size();
     for (int i=;i<n;i++) ans+=st[k][i]->s;
     return ans;
 }

 int getsize(int k)
 {
     int ans=, n=st[k].size();
     for (int i=;i<n;i++) ans+=st[k][i]->rc->s;
     return ans;
 }

 void goleft(int k)
 {
     int n=st[k].size();
     for (int i=;i<n;i++) st[k][i]=st[k][i]->lc;
 }

 void goright(int k)
 {
     int n=st[k].size();
     for (int i=;i<n;i++) st[k][i]=st[k][i]->rc;
 }

 int query(int l,int r,int k)
 {
     if (k>getsum()+getsum()-getsum()-getsum()) return ;
     if (l==r) return l;
     int tmp=getsize()+getsize()-getsize()-getsize();
     int mid=(l+r)>>;
     if (k<=tmp)
     {
         goright(),goright(),goright(),goright();
         return query(mid+,r,k);
     }
     else
     {
         goleft(),goleft(),goleft(),goleft();
         return query(l,mid,k-tmp);
     }
 }

 int lca(int x,int y)
 {
     if (dep[x]>dep[y]) swap(x,y);
     for (int i=low;i>=;i--)
         if (dep[fa[y][i]]>=dep[x]) y=fa[y][i];
     if (x==y) return x;
     for (int i=low;i>=;i--)
         if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
     return fa[x][];
 }

 void write(node *now,int l,int r)
 {
     if (l==r)
     {
         cout<<now->s;
         return ;
     }
     int mid=(l+r)>>;
     write(now->lc,l,mid);
     write(now->rc,mid+,r);
 }

 void dfs(int now)
 {
     ll[now]=rr[now]=++cnt;
     p=sum[fa[now][]];
     sum[now]=change(,num,key[now],);

     for (int j=fir[now];j;j=e[j].next)
     {
         int k=e[j].t;
         if (fa[now][]!=k)
         {
             fa[k][]=now; dep[k]=dep[now]+;
             for (int i=;i<=low;i++) fa[k][i]=fa[fa[k][i-]][i-];
             dfs(k);
             rr[now]=rr[k];
         }
     }
 }

 void add(int x,int y)
 {
     e[++tot].s=x; e[tot].t=y; e[tot].next=fir[x]; fir[x]=tot;
     e[++tot].s=y; e[tot].t=x; e[tot].next=fir[y]; fir[y]=tot;
 }

 int main()
 {
     //freopen("network10.in","r",stdin);
     //freopen("network.out","w",stdout);
     scanf("%d%d",&n,&m);
     low=log(n)/log(2.0);
     for (int i=;i<=n;i++) scanf("%d",&key[i]),w[++num]=key[i];
     int st,ed;
     for (int i=;i<n;i++) scanf("%d%d",&st,&ed),add(st,ed);
     num=n;
     for (int i=;i<=m;i++)
     {
         scanf("%d%d%d",&z[i],&x[i],&y[i]);
         if (!z[i]) w[++num]=y[i];
     }
     sort(w+,w+num+);
     num=unique(w+,w+num+)-w;
     for (int i=;i<=n;i++) key[i]=lower_bound(w+, w+num+, key[i])-w;
     for (int i=;i<=m;i++) if (!z[i]) y[i]=lower_bound(w+, w+num+, y[i])-w;

     sum[]=rot[]=build(,num);
     for (int i=;i<=n;i++) rot[i]=rot[];
     int root=rand()%n+; dep[root]=;
     dfs(root);

     for (int i=;i<=m;i++)
     {
         if (z[i])
         {
             int tmp;
             v[]=x[i],v[]=y[i],v[]=tmp=lca(x[i],y[i]),v[]=fa[tmp][];
             fill(); fill(); fill(); fill();
             int ans=query(,num,z[i]);
             if (ans)
                 printf("%d\n",w[ans]);
             else
                 printf("invalid request!\n");
         }
         else
         {
             adj(ll[x[i]],key[x[i]],-),adj(rr[x[i]]+,key[x[i]],);
             key[x[i]]=y[i];
             adj(ll[x[i]],key[x[i]],),adj(rr[x[i]]+,key[x[i]],-);
         }
     }
     return ;
 }

network