HYSBZ 1036树的统计Count 树链剖分
来源:互联网 发布:淘宝喵喵通信 编辑:程序博客网 时间:2024/04/25 06:22
题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1036
题意:Description
一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. 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”的操作,每行输出一个整数表示要求输出的结果。
思路:树链剖分,都是泪啊,一不小心就写挫啦!!!用线段树同时维护最大值与区间和,加条件判断查询相应的值,简单题。。。
#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int N = 30010;const int INF = 0x3f3f3f3f;struct edge{ int to, next;} g[N*2];struct node{ int l, r, val, sum;} s[N*4];int dep[N], siz[N], son[N], top[N], fat[N], id[N], head[N];int d[N][2], val[N];int n, cnt, num;void add_edge(int v, int u){ g[cnt].to = u; g[cnt].next = head[v]; head[v] = cnt++;}void dfs1(int v, int fa, int d){ dep[v] = d, fat[v] = fa, siz[v] = 1, son[v] = 0; for(int i = head[v]; i != -1; i = g[i].next) { int u = g[i].to; if(u != fa) { dfs1(u, v, d + 1); siz[v] += siz[u]; if(siz[son[v]] < siz[u]) son[v] = u; } }}void dfs2(int v, int tp){ top[v] = tp, id[v] = ++num; if(son[v]) dfs2(son[v], tp); for(int i = head[v]; i != -1; i = g[i].next) { int u = g[i].to; if(u != son[v] && u != fat[v]) dfs2(u, u); }}void push_up(int k){ s[k].val = max(s[k<<1].val, s[k<<1|1].val); s[k].sum = s[k<<1].sum + s[k<<1|1].sum;}void build(int l, int r, int k){ s[k].l = l, s[k].r = r; if(l == r) { s[k].sum = s[k].val = val[l]; return; } int mid = (l + r) >> 1; build(l, mid, k << 1); build(mid + 1, r, k << 1|1); push_up(k);}void update(int v, int c, int k){ if(s[k].l == s[k].r) { s[k].sum = s[k].val = c; return; } int mid = (s[k].l + s[k].r) >> 1; if(v <= mid) update(v, c, k << 1); else update(v, c, k << 1|1); push_up(k);}int seek(int l, int r, int kind, int k){ if(l <= s[k].l && s[k].r <= r) { if(kind == 1) return s[k].val; else return s[k].sum; } int mid = (s[k].l + s[k].r) >> 1; int ans = -INF; if(kind == 2) ans = 0; if(l <= mid) { int tmp = seek(l, r, kind, k << 1); if(kind == 1) ans = max(ans, tmp); else ans += tmp; } if(r > mid) { int tmp = seek(l, r, kind, k << 1|1); if(kind == 1) ans = max(ans, tmp); else ans += tmp; } return ans;}int query(int v, int u, int kind){ int t1 = top[v], t2 = top[u], ans = -INF, tmp; if(kind == 2) ans = 0; while(t1 != t2) { if(dep[t1] < dep[t2]) swap(t1, t2), swap(v, u); tmp = seek(id[t1], id[v], kind, 1); if(kind == 1) ans = max(ans, tmp); else ans += tmp; v = fat[t1]; t1 = top[v]; } if(dep[v] > dep[u]) swap(v, u); tmp = seek(id[v], id[u], kind, 1); if(kind == 1) return ans = max(ans, tmp); else return ans += tmp; return ans;}void slove(){ int m, a, b; char str[100]; scanf("%d", &m); while(m--) { scanf(" %s%d%d", str, &a, &b); if(str[0] == 'C') update(id[a], b, 1); /*更新其在线段树中对应的点,小心啊!!!*/ else { if(str[1] == 'M') printf("%d\n", query(a, b, 1)); else printf("%d\n", query(a, b, 2)); } }}int main(){ while(~ scanf("%d", &n)) { memset(head, -1, sizeof head); cnt = num = 0; for(int i = 1; i <= n - 1; i++) { scanf("%d%d", &d[i][0], &d[i][1]); add_edge(d[i][0], d[i][1]); add_edge(d[i][1], d[i][0]); } dfs1(1, 0, 1); dfs2(1, 1); for(int i = 1; i <= n; i++) scanf("%d", &val[id[i]]); build(1, num, 1); slove(); } return 0;}
0 0
- hysbz 1036 树的统计Count(树链剖分)
- hysbz 1036 树的统计Count 树链剖分
- HYSBZ 1036 树的统计Count(树链剖分)
- HYSBZ 1036 树的统计Count 树链剖分
- HYSBZ 1036树的统计Count 树链剖分
- HYSBZ 1036 树的统计Count
- HYSBZ 1036 [ZJOI2008]树的统计Count
- HYSBZ 1036 树的统计Count
- HYSBZ 1036 树的统计Count
- HYSBZ 1036 树的统计Count (树链剖分 + 线段树)
- HYSBZ 1036 树的统计Count(树链剖分+线段树)
- HYSBZ - 1036 树的统计Count 树链剖分 求和+最大值
- HYSBZ 1036 树的统计Count(树链剖分)
- HYSBZ 1036 树的统计Count(树链剖分)
- HYSBZ 1036 树的统计Count 树链剖分 点权剖分
- 树的统计Count HYSBZ
- 树链剖分点更新模板HYSBZ 1036树的统计Count
- HYSBZ - 1036 树的统计Count(树剖)
- 1090. Highest Price in Supply Chain (25)【树】——PAT (Advanced Level) Practise
- int与Integer的区别
- php &$
- bestcoder百度之星2016AK 1001&1002&1003&1004 题解
- Windows网络编程学习笔记(4) 编写一个TCP客户端
- HYSBZ 1036树的统计Count 树链剖分
- PHP中的可变参数函数和可选参数函数
- 1091. Acute Stroke (30)【搜索】——PAT (Advanced Level) Practise
- 警察与厨师
- 使用hexo搭建博客并上传GitHub
- 1092. To Buy or Not to Buy (20)【水题】——PAT (Advanced Level) Practise
- C++_FatMouse' Trade(贪心)
- 使用.NET Framework提供的InstallUtil.exe安装Windows服务安装时遇到的2种常见错误
- Android并发编程之图文解析volatile关键字