【BZOJ1146】【树链剖分+平衡树】网络管理Network

Description

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

Input

第 一行为两个整数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,此时路径上只有一个路由器。

Output

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

Sample Input

5 5
5 1 2 3 4
3 1
2 1
4 3
5 3
2 4 5
0 1 2
2 2 3
2 1 4
3 3 5

Sample Output

3
2
2
invalid request!

Hint


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

Source

CTSC2008
【分析】
居然1A了,天不负有心人啊.....写了1个半小时TAT
裸的树上带修改第k大,用树链剖分转成线段树,在每个线段树里面维护一个平衡树。
查找的时候先二分答案,再按照树链剖分的方式找就行了。
发现动态内存分配还是蛮好的,均摊分析,空间复杂度不会爆。
  1 /*
  2 唐代韦庄
  3 《菩萨蛮·劝君今夜须沈醉》
  4 劝君今夜须沉醉,尊前莫话明朝事。珍重主人心,酒深情亦深。
  5 须愁春漏短,莫诉金杯满。遇酒且呵呵,人生能几何! 
  6 */
  7 #include <iostream>
  8 #include <cstdio>
  9 #include <algorithm>
 10 #include <cstring>
 11 #include <vector>
 12 #include <utility>
 13 #include <iomanip>
 14 #include <string>
 15 #include <cmath>
 16 #include <queue>
 17 #include <assert.h>
 18 #include <map>
 19 #include <ctime>
 20 #include <cstdlib>
 21 #include <stack>
 22 #include <set> 
 23 #define LOCAL
 24 const int INF = 0x7fffffff;
 25 const int MAXN = 80000  + 10;
 26 const int maxnode = 20000 * 2 + 200000 * 20;
 27 const int MAXM = 1000000 + 10;
 28 const int MAX = 100000000;
 29 using namespace std;
 30 struct Treap{
 31        int val;
 32        int fix, siz;
 33        Treap *ch[2];
 34        
 35        void update(){
 36             siz = 1;
 37             if (ch[0] != NULL) siz += ch[0]->siz;
 38             if (ch[1] != NULL) siz += ch[1]->siz;
 39        }
 40 };
 41 //将t的d儿子换到t 
 42 void rotate(Treap *&t, int d){
 43      Treap *p = t->ch[d];
 44      t->ch[d] = p->ch[d ^ 1];
 45      t->update();
 46      p->ch[d ^ 1] = t;
 47      p->update();
 48      t = p; 
 49      return;
 50 }
 51 Treap *NEW(int val){
 52       Treap *t = new Treap;
 53       t->fix = rand();
 54       t->val = val;
 55       t->siz = 1;
 56       t->ch[0] = t->ch[1] = NULL;
 57       return t;
 58 }
 59 void insert(Treap *&t, int val){
 60      if (t == NULL){
 61         t = NEW(val);
 62         return;
 63      }
 64      int d = (val >= t->val);
 65      insert(t->ch[d], val);
 66      if (t->ch[d]->fix > t->fix) rotate(t, d);
 67      t->update();
 68 }
 69 int size(Treap *&t) {return (t == NULL) ? 0: t->siz;}
 70 //统计比val大的数量,包括val! 
 71 int get(Treap *&t, int val){
 72     if (t == NULL) return 0;
 73     //右边都比他大.. 
 74     if (val <= t->val) return size(t->ch[1]) + 1 + get(t->ch[0], val);
 75     else return get(t->ch[1], val); 
 76 }
 77 void erase(Treap *&t, int x){
 78      int d;
 79      if (x == t->val) d = -1;
 80      else d = (x > t->val);//随意删除一个就行了
 81      
 82      if (d == -1){
 83         Treap *tmp = t;
 84         if (t->ch[0] == NULL){//左儿子为0那么就变成右儿子 
 85            t = t->ch[1];
 86            delete tmp;
 87         }else if (t->ch[1] == NULL){
 88            t = t->ch[0];
 89            delete tmp;
 90         }else{//还没到叶子转下去 
 91            int f = (t->ch[1]->fix > t->ch[0]->fix);
 92            rotate(t, f);
 93            erase(t->ch[f ^ 1], x);//注意转到另一边了 
 94         }
 95      }else erase(t->ch[d], x);
 96      if (t != NULL) t->update();
 97 }
 98 
 99 //------------------------以上为Treap------------------//
100  
101 struct Node{//线段树的节点
102        int l, r;
103        Treap *t;//注意这个里面用动态内存分配...无奈. 
104        Node *ch[2];
105 }*root, mem[MAXN * 2];
106 int n, q;//点数和询问总述数量 
107 int head[MAXN * 2], to[MAXN * 2];//vector会爆? 
108 int M, next[MAXN * 2], fa[MAXN];
109 int Time, data[MAXN], son[MAXN], siz[MAXN], pos[MAXN];//时间轴 
110 int top[MAXN], tot, dep[MAXN];
111 
112 void addEdge(int u, int v){
113      to[M] = v;
114      next[M]= head[u];
115      head[u] = M++;
116 }
117 void dfs_1(int u){
118      son[u] = 0;
119      siz[u] = 1;
120      for (int i = head[u];i != -1; i = next[i]){
121          int v = to[i];
122          if (v == fa[u]) continue;
123          fa[v] = u;
124          dep[v] = dep[u] + 1;
125          dfs_1(v);
126          siz[u] += siz[v];
127          if (siz[v] > siz[son[u]]) son[u] = v;
128      }
129 }
130 void dfs_2(int u, int top_node){
131      pos[u] = ++Time;
132      top[u] = top_node;
133      if (son[u]) dfs_2(son[u], top_node);
134      for (int i = head[u]; i != -1; i = next[i]){
135          int v = to[i];
136          if (v == fa[u] || v == son[u]) continue;
137          dfs_2(v, v); 
138      } 
139 }
140 Node *NEW(int l, int r){
141      Node *p = &mem[tot++];
142      p->l = l;
143      p->r = r;
144      p->t = NULL;//p里面的treap
145      p->ch[0] = p->ch[1] = NULL;
146      return p; 
147 }
148 void build(Node *&t, int l, int r){
149      if (t == NULL){
150         t = NEW(l, r);
151      }
152      if (l == r) return;
153      int mid = (l + r) >> 1;
154      build(t->ch[0], l, mid);
155      build(t->ch[1], mid + 1, r);
156 }
157 //居然是单点修改OAO 
158 //将l位置的路由器延迟时间由x变为y 
159 void change(Node *&p, int l, int x, int y){
160      if (p->l == l && p->r == l){
161         if (x != INF) erase(p->t, x);//为INF的时候就是单纯的插入 
162         insert(p->t, y);
163         return;
164      }
165      if (x != INF) erase(p->t, x);
166      insert(p->t, y);
167      
168      int mid = (p->l + p->r) >> 1;
169      if (l <= mid) change(p->ch[0], l, x, y);
170      else change(p->ch[1], l, x, y); 
171 }
172 //表示在l,r这一段内比val大的数的个数 
173 int query(Node *&p, int l, int r, int val){
174     if (l <= p->l && p->r <= r) return get(p->t, val);
175     int mid = (p->l + p->r) >> 1;
176     int sum = 0;
177     if (l <= mid) sum += query(p->ch[0], l, r, val);
178     if (r > mid) sum += query(p->ch[1], l, r, val); 
179     return sum;
180 }
181 void init(){
182      memset(head, -1, sizeof(head));
183      memset(dep, 0, sizeof(dep));//表示根到自己不包含根的节点数量 
184      siz[0] = fa[1] = 0;
185      M = Time = 0;
186      
187      scanf("%d%d", &n, &q);
188      for (int i = 1; i <= n; i++) scanf("%d", &data[i]);
189      for (int i = 1; i < n; i++){
190          int u, v;
191          scanf("%d%d", &u, &v);
192          addEdge(u, v);
193          addEdge(v, u);
194      }
195      dfs_1(1);
196      dfs_2(1, 1);
197      build(root, 1, Time);
198      for (int i = 1; i <= n; i++) 
199      change(root, pos[i], INF, data[i]);
200 }
201 //返回在lr区间内比x大的数个数 
202 int check(int l, int r, int x){
203     int sum = 0;
204     while (top[l] != top[r]){
205           //保证永远矮的往上爬 
206           if (dep[top[l]] < dep[top[r]]) swap(l, r);
207           sum += query(root, pos[top[l]], pos[l], x);
208           l = fa[top[l]];
209     }
210     //显然l应该高一点 
211     if (dep[l] > dep[r]) swap(l, r);
212     sum += query(root, pos[l], pos[r], x);
213     return sum;
214 }
215 int query(int L, int R, int k){
216     //首先二分最大值k
217     int Ans, l = -MAX, r = MAX;
218     while (l <= r){
219           int mid = (l + r) >> 1;
220           if (mid == 3)
221           printf("");
222           if (check(L, R, mid) >= k) Ans = mid, l = mid + 1;
223           else r = mid - 1; 
224     }
225     //printf("%d", check(L, R, 3));
226     return Ans;
227 }
228 int search(int l, int r){
229     int sum = 0;
230     while (top[l] != top[r]){
231           if (dep[top[l]] < dep[top[r]]) swap(l, r);
232           sum += dep[l] - dep[top[l]] + 1;
233           l = fa[top[l]];
234     }
235     //dep[r]大 
236     if (dep[l] > dep[r]) swap(l, r);
237     sum += dep[r] - dep[l] + 1;
238     return sum;
239 } 
240 void work(){
241      for (int i = 1; i <= q; i++){
242         
243          int l, r, k;
244          scanf("%d%d%d", &k, &l, &r); 
245          //if (i == 5)
246          //printf("%d", k);
247          if (k == 0){//修改操作 
248             change(root, pos[l], data[l], r);
249             data[l] = r;
250          }else{//查询
251             int num = search(l, r);//查询l到r之间的节点个数 
252             if (k > num) {printf("invalid request!\n");continue;}
253             printf("%d\n", query(l, r, k));
254          }
255      }
256 }
257 void debug(){
258      /*Treap *root = NULL;
259      insert(root, 2);
260      printf("%d", get(root, 2));*/ 
261      //printf("%d", search(4, 2));
262      //for (int i = 1; i <= 5; i++) printf("%d\n", pos[i]);
263      //printf("%d", query(root, 2, 3, 3));
264 }
265 
266 int main(){
267     //srand(time(0));
268     
269     init();
270     work();
271     //debug();
272     return 0;
273 }
View Code

 

 
posted @ 2015-03-17 19:19  TCtower  阅读(219)  评论(0编辑  收藏  举报