[BZOJ 1146] [CTSC2008]网络管理Network（树状数组+主席树）

题目描述

M公司是一个非常庞大的跨国公司，在许多国家都设有它的下属分支机构或部门。为了让分布在世界各地的N个部门之间协同工作，公司搭建了一个连接整个公司的通信网络。该网络的结构由N个路由器和N-1条高速光缆组成。每个部门都有一个专属的路由器，部门局域网内的所有机器都联向这个路由器，然后再通过这个通信子网与其他部门进行通信联络。该网络结构保证网络中的任意两个路由器之间都存在一条直接或间接路径以进行通信。高速光缆的数据传输速度非常快，以至于利用光缆传输的延迟时间可以忽略。但是由于路由器老化，在这些路由器上进行数据交换会带来很大的延迟。而两个路由器之间的通信延迟时间则与这两个路由器通信路径上所有路由器中最大的交换延迟时间有关。作为M公司网络部门的一名实习员工，现在要求你编写一个简单的程序来监视公司的网络状况。该程序能够随时更新网络状况的变化信息（路由器数据交换延迟时间的变化），并且根据询问给出两个路由器通信路径上延迟第k大的路由器的延迟时间。

【任务】你的程序从输入文件中读入N个路由器和N-1条光缆的连接信息，每个路由器初始的数据交换延迟时间Ti，以及Q条询问（或状态改变）的信息。并依次处理这Q条询问信息，它们可能是：

由于更新了设备，或者设备出现新的故障，使得某个路由器的数据交换延迟时间发生了变化。
查询某两个路由器a和b之间的路径上延迟第k大的路由器的延迟时间。

输入输出格式

输入格式：

第一行为两个整数N和Q，分别表示路由器总数和询问的总数。

第二行有N个整数，第i个数表示编号为i的路由器初始的数据延迟时间Ti。

紧接着N-1行，每行包含两个整数x和y。表示有一条光缆连接路由器x和路由器y。

紧接着是Q行，每行三个整数k、a、b。

如果k=0，则表示路由器a的状态发生了变化，它的数据交换延迟时间由Ta变为b。

如果k>0，则表示询问a到b的路径上所经过的所有路由器（包括a和b）中延迟第k大的路由器的延迟时间。注意a可以等于b，此时路径上只有一个路由器。

输出格式：

对于每一个第二种询问（k>0），输出一行。包含一个整数为相应的延迟时间。如果路径上的路由器不足k个，则输出信息“invalid request!”（全部小写不包含引号，两个单词之间有一个空格）。

输入输出样例

输入样例#1：复制

输出样例#1：复制

3

2

2

invalid request!

说明

测试数据满足N,Q<=80000,任意一个路由器在任何时刻都满足延迟时间小于10^8。对于所有询问满足0<=K<=N 。

题解

　　带修树链第k大

　　据说能用树剖+线段树套平衡树4个log稳过？

　　据说能用树剖+树状数组套主席树3个log？

　　据说能把树剖去了两个log？

　　据说还能一个log？

　　蒟蒻表示很懵逼……只会用树状数组套主席树的……

　　首先这是一棵树，要在树上跑主席树两种方法，一种是每一次在父亲的树的基础上加值，然而那样就不资瓷修改了（大概？蒟蒻也不是很清楚）

　　或者用括号序列的思想，每一个点dfs进的时候存个点，出去的时候存个点，这样就可以用差分来跑主席树了（具体细节看代码）

　　然后是修改……我就当你们都会待修改主席树了（不懂的可以去看看这道题->动态逆序对）

　　然后……大概就这样？

　　时间复杂度应该是两个log吧（树剖用来求LCA的应该不算在里面）

　　ps：调主席树update的时候写成$ch[x][0]$和$ch[x][1]$了……调了半天啥错都找不到orz

 //minamoto

 #include<iostream>

 #include<cstdio>

 #include<algorithm>

 using namespace std;

 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)

 char buf[<<],*p1=buf,*p2=buf;

 inline int read(){

     #define num ch-'0'

     char ch;bool flag=;int res;

     while(!isdigit(ch=getc()))

     (ch=='-')&&(flag=true);

     for(res=num;isdigit(ch=getc());res=res*+num);

     (flag)&&(res=-res);

     #undef num

     return res;

 }

 const int N=;

 struct node{

     int op,x,y;

 }Q[N];

 int n,m,q,b[N<<],lim;

 int rt[N<<],L[N<<],R[N<<],sz[N<<],cnt;

 void update(int &p,int l,int r,int pos,int val){

     if(!p) p=++cnt;sz[p]+=val;

     if(l==r) return;

     int mid=l+r>>;

     if(pos<=mid) update(L[p],l,mid,pos,val);

     else update(R[p],mid+,r,pos,val);

 }

 void update(int x,int val,int op){

     for(;x<=m;x+=x&-x) update(rt[x],,lim,val,op);

 }

 int ver[N<<],Next[N<<],head[N],tot;

 int fa[N],dep[N],top[N],size[N],son[N],ls[N],rs[N],sum[N],val[N],num;

 inline void add(int u,int v){

     ver[++tot]=v,Next[tot]=head[u],head[u]=tot;

     ver[++tot]=u,Next[tot]=head[v],head[v]=tot;

 }

 void dfs1(int u){

     size[u]=,dep[u]=dep[fa[u]]+;

     for(int i=head[u];i;i=Next[i]){

         int v=ver[i];

         if(v!=fa[u]){

             fa[v]=u,dfs1(v),size[u]+=size[v];

             if(size[v]>size[son[u]]) son[u]=v;

         }

     }

 }

 void dfs2(int u,int t){

     top[u]=t,ls[u]=++num;

     update(ls[u],val[u],);

     if(son[u]){

         dfs2(son[u],t);

         for(int i=head[u];i;i=Next[i]){

             int v=ver[i];

             if(v!=fa[u]&&v!=son[u]) dfs2(v,v);

         }

     }

     rs[u]=++num;

     update(rs[u]+,val[u],-);

 }

 int LCA(int u,int v){

     while(top[u]!=top[v]){

         if(dep[top[u]]<dep[top[v]]) swap(u,v);

         u=fa[top[u]];

     }

     return dep[u]<dep[v]?u:v;

 }

 int sta[],stb[],cnta,cntb;

 inline void get(int x,int op){

     if(~op){

         for(;x;x-=x&-x) sta[++cnta]=rt[x];

     }

     else{

         for(;x;x-=x&-x) stb[++cntb]=rt[x];

     }

 }

 int query(int l,int r,int k){

     if(l==r) return b[l];

     int mid=l+r>>,sum=;

     for(int i=;i<=cnta;++i) sum+=sz[R[sta[i]]];

     for(int i=;i<=cntb;++i) sum-=sz[R[stb[i]]];

     if(sum>=k){

         for(int i=;i<=cnta;++i) sta[i]=R[sta[i]];

         for(int i=;i<=cntb;++i) stb[i]=R[stb[i]];

         return query(mid+,r,k);

     }

     else{

         for(int i=;i<=cnta;++i) sta[i]=L[sta[i]];

         for(int i=;i<=cntb;++i) stb[i]=L[stb[i]];

         return query(l,mid,k-sum);

     }

 }

 inline void getans(int u,int v,int k){

     int lca=LCA(u,v);

     if(k>dep[u]+dep[v]-(dep[lca]<<)+) return (void)(puts("invalid request!"));

     cnta=cntb=;

     get(ls[u],),get(ls[lca],-);

     get(ls[v],),get(ls[fa[lca]],-);

     printf("%d\n",query(,lim,k));

 }

 inline void change(int pos,int v){

     update(ls[pos],val[pos],-),update(rs[pos]+,val[pos],),val[pos]=v;

     update(ls[pos],v,),update(rs[pos]+,v,-);

 }

 int main(){

     //freopen("testdata.in","r",stdin);

     n=read(),q=read(),m=n<<;

     for(int i=;i<=n;++i) b[++lim]=val[i]=read();

     for(int i=;i<n;++i){

         int u=read(),v=read();add(u,v);

     }

     for(int i=;i<=q;++i){

         int op=read(),x=read(),y=read();

         Q[i]=(node){op,x,y};

         if(!op) b[++lim]=y;

     }

     sort(b+,b++lim),lim=unique(b+,b++lim)-b-;

     for(int i=;i<=n;++i) val[i]=lower_bound(b+,b++lim,val[i])-b;

     dfs1(),dfs2(,);

     for(int i=;i<=q;++i){

         int op=Q[i].op,x=Q[i].x,y=Q[i].y;

         if(op) getans(x,y,op);

         else y=lower_bound(b+,b++lim,y)-b,change(x,y);

     }

     return ;

 }