HYSBZ 1036 树的统计Count 树链剖分
2016-05-14 20:27
323 查看
题目: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”的操作,每行输出一个整数表示要求输出的结果。
思路:树链剖分,都是泪啊,一不小心就写挫啦!!!用线段树同时维护最大值与区间和,加条件判断查询相应的值,简单题。。。
题意: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 , siz , son , top , fat , id , head ; int d [2], val ; 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; }
相关文章推荐
- Git与Repo入门
- 第十周项目5——摩托车继承自行车和机动车
- SSH网上答题系统质量属性
- 1093. Count PAT's (25)【计数】——PAT (Advanced Level) Practise
- 说反话
- QT-如何添加资源文件
- 一个屌丝程序猿的人生(八)
- Android并发编程之图文解析volatile关键字
- C++_FatMouse' Trade(贪心)
- mysql5.7下的timestampn Error : Invalid default value for 'timestamp'
- 如何用java实现使用电子邮件控制你的电脑
- 在fragment中获取他附着的activity中的变量
- 1092. To Buy or Not to Buy (20)【水题】——PAT (Advanced Level) Practise
- Windows网络编程学习笔记(4) 编写一个TCP客户端
- 警察与厨师
- 1091. Acute Stroke (30)【搜索】——PAT (Advanced Level) Practise
- Nginx 安装升级脚本
- volatile实现可见性但不保证原子性
- 使用hexo搭建博客并上传GitHub
- Codeforces Round #349 (Div. 1) C E (计数. SAM+线段树)