【BZOJ1036】【树链剖分】树的统计Count

Description

一 棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 III. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意:从点u到点v的路径上的节点包括u和v本身

Input

输 入的第一行为一个整数n,表示节点的个数。接下来n – 1行,每行2个整数a和b,表示节点a和节点b之间有一条边相连。接下来n行,每行一个整数,第i行的整数wi表示节点i的权值。接下来1行,为一个整数 q,表示操作的总数。接下来q行,每行一个操作,以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 对于100%的数据,保证1<=n<=30000,0<=q<=200000;中途操作中保证每个节点的权值w在-30000到 30000之间。

Output

对于每个“QMAX”或者“QSUM”的操作,每行输出一个整数表示要求输出的结果。

Sample Input

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output

4
1
2
2
10
6
5
6
5
16

HINT

Source

树的分治

【分析】

SB题,不说了。

  1 /*
  2 唐代李白
  3 《南陵别儿童入京》
  4 白酒新熟山中归,黄鸡啄黍秋正肥。
  5 呼童烹鸡酌白酒,儿女嬉笑牵人衣。
  6 高歌取醉欲自慰,起舞落日争光辉。
  7 游说万乘苦不早,著鞭跨马涉远道。
  8 会稽愚妇轻买臣,余亦辞家西入秦。
  9 仰天大笑出门去,我辈岂是蓬蒿人。
 10 */
 11 #include <iostream>
 12 #include <cstdio>
 13 #include <algorithm>
 14 #include <cstring>
 15 #include <vector>
 16 #include <utility>
 17 #include <iomanip>
 18 #include <string>
 19 #include <cmath>
 20 #include <queue>
 21 #include <assert.h>
 22 #include <map>
 23 #include <ctime>
 24 #include <cstdlib>
 25 #include <stack>
 26 #define LOCAL
 27 const int INF = 0x3f3f3f3f;
 28 const int MAXN = 100000  + 10;
 29 const int maxnode = 1000000;
 30 const int maxm= 30000 * 2 + 10;
 31 using namespace std;
 32 struct Node{
 33        int l, r;
 34        int Max, sum;
 35        Node *ch[2];
 36 }*root, mem[maxnode];
 37 struct Edge{
 38        int u, v;
 39 }edge[MAXN];
 40
 41 int n, top[MAXN], fa[MAXN];
 42 int dep[MAXN], size[MAXN], son[MAXN];
 43 int head[MAXN], next[MAXN], to[MAXN];
 44 int pos[MAXN], Time, M, tot;
 45
 46 void dfs_1(int u){
 47      size[u] = 1;
 48      son[u] = 0;
 49      for (int i = head[u]; i != -1; i = next[i]){
 50          int v = to[i];
 51          if (v == fa[u]) continue;
 52          fa[v] = u;
 53          dep[v] = dep[u] + 1;
 54          dfs_1(v);
 55          size[u] += size[v];
 56          if (size[v] > size[son[u]]) son[u] = v;
 57      }
 58 }
 59 //标号
 60 void dfs_2(int u, int top_node){
 61      top[u] = top_node;
 62      pos[u] = ++Time;//这里就不是代表边了,而是点
 63      if (son[u]) dfs_2(son[u], top_node);
 64
 65      for (int i = head[u]; i != -1; i = next[i]){
 66          int v = to[i];
 67          if (v == fa[u] || v == son[u]) continue;
 68          dfs_2(v, v);
 69      }
 70 }
 71 void addEdge(int u, int v){
 72      to[M] = v;
 73      next[M] = head[u];
 74      head[u] = M++;
 75 }
 76 Node *NEW(int l, int r){
 77      Node *p = &mem[tot++];
 78      p->l = l;
 79      p->r = r;
 80      p->Max = p->sum = 0;
 81      return p;
 82 }
 83 //连区间修改都没有
 84 void update(Node *&t){
 85      if (t->l == t->r) return;
 86      t->Max = max(t->ch[0]->Max, t->ch[1]->Max);
 87      t->sum = t->ch[1]->sum + t->ch[0]->sum;
 88 }
 89 void build(Node *&t, int l, int r){
 90      if (t == NULL){
 91         t = NEW(l, r);
 92      }
 93      if (l == r) return;
 94      int mid = (l + r) >> 1;
 95      build(t->ch[0], l, mid);
 96      build(t->ch[1], mid + 1, r);
 97 }
 98 void change(Node *&t, int l, int x){
 99      if (t->l == l && t->r == l){
100         t->sum = t->Max = x;
101         return;
102      }
103      int mid = (t->l + t->r) >> 1;
104      if (l <= mid) change(t->ch[0], l, x);
105      else change(t->ch[1], l, x);
106
107      update(t);
108 }
109 void init(){
110      memset(dep, 0, sizeof(dep));
111      memset(head, -1, sizeof(head));
112
113      M = Time = tot = 0;
114      scanf("%d", &n);
115      for (int i = 1; i < n; i++){
116          scanf("%d%d", &edge[i].u, &edge[i].v);
117          addEdge(edge[i].u, edge[i].v);
118          addEdge(edge[i].v, edge[i].u);
119      }
120      fa[1] = 0;
121      size[0] = 0;
122
123      dfs_1(1);
124      dfs_2(1, 1);
125      root = NULL;
126      build(root, 1, Time);
127      //因为不是边所以跟父亲没什么关系,不用对边进行更新
128      for (int i = 1; i <= n; i++){
129          int t;
130          scanf("%d", &t);
131          change(root, pos[i], t);
132      }
133      //printf("%d\n", root->Max);
134 }
135 int query_sum(Node *&t, int l, int r){
136     if (l <= t->l && t->r <= r) return t->sum;
137     int mid = (t->l + t->r) >> 1;
138     int cnt = 0;
139     if (l <= mid) cnt += query_sum(t->ch[0], l, r);
140     if (r > mid) cnt += query_sum(t->ch[1], l, r);
141     return cnt;
142 }
143 int query_max(Node *&t, int l, int r){
144     if (l <= t->l && t->r <= r) return t->Max;
145     int mid = (t->l + t->r) >> 1;
146     int Ans = -INF;
147     if (l <= mid) Ans = max(Ans, query_max(t->ch[0], l, r));
148     if (r > mid) Ans = max(Ans, query_max(t->ch[1], l, r));
149     return Ans;
150 }
151 int QMAX(int l, int r){
152      int Ans = -INF;
153      while (top[l] != top[r]){
154            //低的往上爬
155            if (dep[top[l]] < dep[top[r]]) swap(l, r);
156            Ans = max(Ans, query_max(root, pos[top[l]], pos[l]));
157            l = fa[top[l]];
158      }
159      //if (l == r) return Ans;
160      if (dep[l] > dep[r]) swap(l, r);
161      Ans = max(Ans, query_max(root, pos[l], pos[r]));
162      return Ans;
163 }
164 int QSUM(int l, int r){
165      int sum = 0;
166      while (top[l] != top[r]){
167            //低的往上爬
168            if (dep[top[l]] < dep[top[r]]) swap(l, r);
169            sum += query_sum(root, pos[top[l]], pos[l]);
170            l = fa[top[l]];
171      }
172      //if (l == r) return sum;
173      if (dep[l] > dep[r]) swap(l, r);
174      sum += query_sum(root, pos[l], pos[r]);
175      return sum;
176 }
177 void work(){
178      int q;
179      scanf("%d", &q);
180      for (int i = 1; i <= q; i++){
181          char str[10];
182          scanf("%s", str);
183          if (str[0] == ‘C‘){
184             int l, x;
185             scanf("%d%d", &l, &x);
186             change(root, pos[l], x);
187          }else{
188                if (str[1] == ‘M‘){
189                   int l, r;
190                   scanf("%d%d", &l, &r);
191                   printf("%d\n", QMAX(l, r));
192                }else{
193                   int l, r;
194                   scanf("%d%d", &l, &r);
195                   printf("%d\n", QSUM(l, r));
196                }
197          }
198      }
199 }
200
201 int main(){
202     int T;
203
204     init();
205     work();
206     return 0;
207 }

时间: 2024-10-25 16:41:55

【BZOJ1036】【树链剖分】树的统计Count的相关文章

hdu 3966 Aragorn&#39;s Story(树链剖分+树状数组)

题目链接:hdu 3966 Aragorn's Story 题目大意:给定一个棵树,然后三种操作 Q x:查询节点x的值 I x y w:节点x到y这条路径上所有节点的值增加w D x y w:节点x到y这条路径上所有节点的值减少w 解题思路:树链剖分,用树状数组维护每个节点的值. #pragma comment(linker, "/STACK:1024000000,1024000000") #include <cstdio> #include <cstring>

树链剖分 树的统计

/*题目描述 Description一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w. 我们将以下面的形式来要求你对这棵树完成一些操作: I.                    CHANGE u t : 把结点u的权值改为tII.                 QMAX u v: 询问从点u到点v的路径上的节点的最大权值III.               QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意:从点u到点v的路径上的节点包括u和v本身 输入描述 In

Codeforces Round #425 (Div. 2) Problem D Misha, Grisha and Underground (Codeforces 832D) - 树链剖分 - 树状数组

Misha and Grisha are funny boys, so they like to use new underground. The underground has n stations connected with n - 1 routes so that each route connects two stations, and it is possible to reach every station from any other. The boys decided to h

bzoj1146整体二分+树链剖分+树状数组

其实也没啥好说的 用树状数组可以O(logn)的查询 套一层整体二分就可以做到O(nlngn) 最后用树链剖分让序列上树 1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 #include<algorithm> 5 using namespace std; 6 inline int read() 7 { 8 int x=0,f=1,ch=getchar(); 9 while(ch<

HDU 5044 (树链剖分+树状数组+点/边改查)

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=5044 题目大意:修改链上点,修改链上的边.查询所有点,查询所有边. 解题思路: 2014上海网赛的变态树链剖分模板题.将以往树链剖分的点&边修改和查询合在一起之后,难度上去不少. 第一个卡人点是读入优化. 第二个卡人点是树状数组.由于要查询所有点,如果使用线段树,每次都要扫到底层才能取出点值,必T无疑. 然后使用树状数组之后,树链剖分的点/边修改写法有些变动. 点查询变化不大. 边查询只要查询一下

BZOJ 1146: [CTSC2008]网络管理Network( 树链剖分 + 树状数组套主席树 )

树链剖分完就成了一道主席树裸题了, 每次树链剖分找出相应区间然后用BIT+(可持久化)权值线段树就可以完成计数. 但是空间问题很严重....在修改时不必要的就不要新建, 直接修改原来的..详见代码. 时间复杂度O(N*log^3(N)) ---------------------------------------------------------------------------- #include<cstdio> #include<cstring> #include<

POJ 2763 Housewife Wind(树链剖分+树状数组)

[题目链接] http://poj.org/problem?id=2763 [题目大意] 在一棵树上,给出一些边的边长,有修改边的边长的操作, 询问每次从当前点到目标点的最短距离 [题解] 树链剖分之后,相当于树状数组的单点更新和区间查询, 注意边权转点权之后链操作不覆盖deep最浅的点,这里容易出错 [代码] #include <cstdio> #include <cstring> #include <algorithm> using namespace std; c

【LuoguP3038/[USACO11DEC]牧草种植Grass Planting】树链剖分+树状数组【树状数组的区间修改与区间查询】

模拟题,可以用树链剖分+线段树维护. 但是学了一个厉害的..树状数组的区间修改与区间查询.. 分割线里面的是转载的: -------------------------------------------------------------------------------- [ 3 ]  上面都不是重点--重点是树状数组的区间修改+区间查询 这个很好玩 其实也挺简单 首先依旧是引入delta数组 delta[i]表示区间 [i, n] 的共同增量 于是修改区间 [l, r] 时修改 delt

(简单) POJ 3321 Apple Tree,树链剖分+树状数组。

Description There is an apple tree outside of kaka's house. Every autumn, a lot of apples will grow in the tree. Kaka likes apple very much, so he has been carefully nurturing the big apple tree. The tree has N forks which are connected by branches.

Codeforces Round #425 (Div. 2) D 树链剖分 + 树状数组维护区间

一看就知道 可以LCA判断做 也可以树链剖分拿头暴力 然而快速读入和线段树维护区间会T70 于是只能LCA? 线段树的常数不小 于是需要另外一种办法来进行区间加减和查询区间和 就是使用树状数组 这个题的代码 #include<iostream> #include<cstdio> #include<cmath> #include<algorithm> #include<vector> #include<cstring> using na