ICPC 南昌现场赛 K：Tree（dsu on tree + 动态开点线段树）

#include <bits/stdc++.h>using namespace std;const int N = 2e5 + 10;int head[N], to[N], nex[N], cnt = 1;int value[N], n, m;int son[N], sz[N], dep[N], l[N], r[N], rk[N], tot;int root[N], ls[N << 6], rs[N << 6], sum[N << 6], num;void add(int x, int y) {to[cnt] = y;nex[cnt] = head[x];head[x] = cnt++;
}void dfs(int rt, int fa) {dep[rt] = dep[fa] + 1, sz[rt] = 1, l[rt] = ++tot, rk[tot] = rt;for (int i = head[rt]; i; i = nex[i]) {if (to[i] == fa) {continue;}dfs(to[i], rt);sz[rt] += sz[to[i]];if (!son[rt] || sz[son[rt]] < sz[to[i]]) {son[rt] = to[i];}}r[rt] = tot;
}void push_up(int rt) {sum[rt] = sum[ls[rt]] + sum[rs[rt]];
}void update(int &rt, int l, int r, int x, int value) {if (!rt) {rt = ++num;}if (l == r) {sum[rt] += value;return ;}int mid = l + r >> 1;if (x <= mid) {update(ls[rt], l, mid, x, value);}else {update(rs[rt], mid + 1, r, x, value);}push_up(rt);
}int query(int rt, int l, int r, int L, int R) {if (!rt) {return 0;}if (l >= L && r <= R) {return sum[rt];}int mid = l + r >> 1, ans = 0;if (L <= mid) {ans += query(ls[rt], l, mid, L, R);}if (R > mid) {ans += query(rs[rt], mid + 1, r, L, R);}return ans;
}long long ans;void dfs(int rt, int fa, bool keep) {for (int i = head[rt]; i; i = nex[i]) {if (to[i] == fa || to[i] == son[rt]) {continue;}dfs(to[i], rt, 0);}if (son[rt]) {dfs(son[rt], rt, 1);}int v = 2 * value[rt], d = dep[rt];for (int i = head[rt]; i; i = nex[i]) {if (to[i] == fa || to[i] == son[rt]) {continue;}for (int j = l[to[i]]; j <= r[to[i]]; j++) {int target_v = v - value[rk[j]], last_d = m - (dep[rk[j]] - d);//目标权值，剩下的可延展的距离if (target_v < 0 || last_d <= 0) {//如果目标权值小于0或者剩下的可延展距离没有了，提前剪除不合法continue;}int l = d + 1, r = d + last_d;//深度的区间范围，然后查询即可。ans += query(root[target_v], 1, n,  l, r);}for (int j = l[to[i]]; j <= r[to[i]]; j++) {update(root[value[rk[j]]], 1, n, dep[rk[j]], 1);}}update(root[value[rt]], 1, n, dep[rt], 1);if (!keep) {for (int i = l[rt]; i <= r[rt]; i++) {update(root[value[rk[i]]], 1, n, dep[rk[i]], -1);}}
}int main() {// freopen("in.txt", "r", stdin);// freopen("out.txt", "w", stdout);// ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);scanf("%d %d", &n, &m);for (int i = 1; i <= n; i++) {scanf("%d", &value[i]);}for (int i = 2; i <= n; i++) {int x;scanf("%d", &x);add(x, i);}dfs(1, 0);dfs(1, 0, 1);printf("%lld\n", ans * 2);return 0;
}

ICPC 南昌现场赛 K：Tree（dsu on tree + 动态开点线段树）相关推荐

2019 ICPC Asia Nanchang Regional K.Tree 树上启发式合并 + 动态开点线段树
传送门文章目录题意: 思路: 题意: 给你一棵树,每个点都有一个权值valvalval,求满足以下条件 (1)x!=yx!=yx!=y (2)xxx和yyy不互为祖先 (3)val[lca(x,y ...
树上启发式合并问题 ---- 2019icpc南昌 K. Tree （树上启发式合并 + 动态开点线段树)
题目链接题目大意: 就是给你一颗树,每个点有个权值viv_ivi,问你有多少对(x,y)(x,y)(x,y)满足: xxx不是yyy的祖先 yyy也不是xxx的祖先 xxx和yyy的距离不超过kk ...
ACM-ICPC 2018 沈阳赛区现场赛 E. The Kouga Ninja Scrolls （切比雪夫距离+线段树）
题目链接: 题意:在二维平面上有 n 个人,每个人有一个位置(xi, yi)和门派 ci,m 个操作:①改变第 k 个人的位置:②改变第 k 个人的门派:③询问区间[l,r]之间不同门派的两个人的最大 ...
bzoj4605 崂山白花蛇草水（动态开点线段树逃替罪羊重构K-D tree）
首先我们发现它是一个三维的问题而且还强制在线囧我们可以考虑在外面来一个权值线段树,然后每次查询时在线段树二分即可,这样我们每个节点再套一棵K-D tree,用来查询有多少个点被框住了,然后因为是不 ...
2019 ICPC 南昌网络赛 H. The Nth Item
2019 ICPC 南昌网络赛 H. The Nth Item 题目大意:已知一个数列F(n): F(0)=0,F(1)=1 F(n)=3∗F(n−1)+2∗F(n−2),(n≥2) 给你一个操作 ...
2018 icpc 徐州现场赛G-树上差分+组合数学-大佬的代码
现场赛大佬打印的代码,观摩了一哈. 写了注释,贴一下,好好学习.%%%PKU 代码: 1 //树上差分(LCA) 2 #include<bits/stdc++.h> 3 4 #define ...
2011 ACM/ICPC 北京现场赛 B HouYi's secret
周日起床后看见QQ群消息说 bupt上有半同步北京现场赛试题,就进入看了下题,一共读了两个题 A和B(因为这两个题目当时就都提交的很多) A题不会做,有限制的生成树的这类知识好像还没有接触过... B ...
2019年ACM-ICPC - 南昌网络赛I：Yukino With Subinterval【带修主席树】
题目: 2019ICPC南昌网络赛I:Yukino With Subinterval 题意: 给定长度为 N 的数组,有两种操作:(1)单点修改,(2)查询区间[qL,qR]内有多少个不同的段(连续相 ...
【bzoj3065】: 带插入区间K小值详解——替罪羊套函数式线段树
不得不说,做过最爽的树套树---- 由于有了区间操作,我们很容易把区间看成一棵平衡树,对他进行插入,那么外面一层就是平衡树了,这就与我们之前所见到的不同了.我们之前所见到的大多数是线段树套平衡树而此题 ...

ICPC 南昌现场赛 K：Tree（dsu on tree + 动态开点线段树）

Tree

ICPC 南昌现场赛 K：Tree（dsu on tree + 动态开点线段树）相关推荐

最新文章

热门文章