E. Party Company

容易發現這是一顆樹形結構，根節點為$1$，并且有點權從根節點開始遞減。

題目大意就是給定一個點$u,l,r$，對于于$u$在同一個連通塊里，并且點權是在$[l,r]$之間的點答案貢獻加一。

如果理解到上述的題意，那這題就變得簡單了。

由于我們要求的是在同一個聯通塊里的，并且點權具有單調性，我們可以通過二分跳轉到可以滿足的深度最小的節點上去，在這個節點依附上$l$。

容易想到，滿足要求的點一定是出現在這個節點的子樹上的，所以我們可以動態維護一個以$l$的有序數組，然后在每個節點二分查找，有多少個值是小于等于當前節點的，這個值就是當前節點的答案了。

當我們退出這顆子樹的時候，記得清空這個節點上依附的$l$。

由于$l,r$都比較小，所以可以直接通過樹狀數組，進行單點修改，前綴和查詢，整體復雜度$O(n\log m+m\log m)$。

#include <bits/stdc++.h>using namespace std;const int N = 1e5 + 10;int head[N], to[N], nex[N], cnt = 1;int fa[N], son[N], sz[N], top[N], rk[N], id[N], dep[N], tot;int n, m, value[N], ans[N];vector<int> vt[N];void add(int x, int y) {to[cnt] = y;nex[cnt] = head[x];head[x] = cnt++; }void dfs1(int rt, int f) {fa[rt] = f, sz[rt] = 1, dep[rt] = dep[f] + 1;for (int i = head[rt]; i; i = nex[i]) {if (to[i] == f) {continue;}dfs1(to[i], rt);sz[rt] += sz[to[i]];if (!son[rt] || sz[to[i]] > sz[son[rt]]) {son[rt] = to[i];}} }void dfs2(int rt, int tp) {top[rt] = tp, rk[++tot] = rt, id[rt] = tot;if (!son[rt]) {return ;}dfs2(son[rt], tp);for (int i = head[rt]; i; i = nex[i]) {if (to[i] == fa[rt] || to[i] == son[rt]) {continue;}dfs2(to[i], to[i]);} }void solve(int cur, int L, int R) {while (cur != 1) {if (fa[top[cur]] && value[fa[top[cur]]] <= R) {cur = fa[top[cur]];}else {int l = id[top[cur]], r = id[cur];while (l < r) {int mid = l + r >> 1;if (value[rk[mid]] > R) {l = mid + 1;}else {r = mid;}}vt[rk[l]].push_back(L);return ;}}vt[1].push_back(L); }int tree[N];int lowbit(int x) {return x & (-x); }void update(int x, int value) {while (x < N) {tree[x] += value;x += lowbit(x);} }int query(int x) {int ans = 0;while (x) {ans += tree[x];x -= lowbit(x);}return ans; }void dfs(int rt, int fa) {for (auto it : vt[rt]) {update(it, 1);}ans[rt] = query(value[rt]);for (int i = head[rt]; i; i = nex[i]) {if (to[i] == fa) {continue;}dfs(to[i], rt);}for (auto it : vt[rt]) {update(it, -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, fa; i <= n; i++) {scanf("%d %d", &value[i], &fa);if (i != 1) {add(fa, i);}}dfs1(1, 0);dfs2(1, 1);for (int i = 1; i <= m; i++) {int cur, l, r;scanf("%d %d %d", &cur, &l, &r);solve(cur, l, r);}dfs(1, 0);for (int i = 1; i <= n; i++) {printf("%d%c", ans[i], i == n ? '\n' : ' ');}return 0; }

總結

以上是生活随笔為你收集整理的E. Party Company（树上问题）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。