华华和月月种树
题目地址:
基本思路:
我们考虑先离线,将树建出来,然后在序上用差分树状数组维护,
但是在这样做的过程中,对于每次的操作二,有些实际还没有被加入的点也被算了权值,
所以我们在每次新加点的时候记录一下,这个点被加进来的时候已经有多少权值了,
很明显这部分权值是无效的,我们在之后的查询里去除这部分就是了。
参考代码:
#pragma GCC optimize(2) #pragma GCC optimize(3) #include <bits/stdc++.h> using namespace std; #define IO std::ios::sync_with_stdio(false); cin.tie(0) #define int long long #define SZ(x) ((int)(x).size()) #define all(x) (x).begin(), (x).end() #define rep(i, l, r) for (int i = l; i <= r; i++) #define per(i, l, r) for (int i = l; i >= r; i--) #define mset(s, _) memset(s, _, sizeof(s)) #define pb push_back #define pii pair <int, int> #define mp(a, b) make_pair(a, b) #define INF 0x3f3f3f3f inline int read() { int x = 0, neg = 1; char op = getchar(); while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); } while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); } return neg * x; } inline void print(int x) { if (x < 0) { putchar('-'); x = -x; } if (x >= 10) print(x / 10); putchar(x % 10 + '0'); } const int maxn = 4e5 + 10; int m,op[maxn],pos[maxn],a[maxn],w[maxn]; int L[maxn],R[maxn],tot; vector<int> G[maxn]; void dfs(int u,int par){ L[u] = ++tot; for(auto to : G[u]){ if(to == par) continue; dfs(to,u); } R[u] = tot; } struct BIT{ int t[maxn << 2]; int lowbit(int x){ return x & (-x); } void clear(){ memset(t,0,sizeof(t)); } int sum(int x){ int res = 0; while(x){ res += t[x]; x -= lowbit(x); } return res; } void update(int x,int v){ while(x <= tot + 1){ t[x] += v; x += lowbit(x); } } }bit; signed main() { m = read(); int cnt = 0; rep(i, 1, m) { op[i] = read(), pos[i] = read(); if (op[i] == 2) a[i] = read(); else if (op[i] == 1) { G[pos[i]].pb(++cnt); G[cnt].push_back(pos[i]); a[i] = cnt; // 记录一下这次加入的是哪个点; } } dfs(0, -1); bit.clear(); for (int i = 1; i <= m; i++) { if (op[i] == 1) { w[L[a[i]]] = bit.sum(L[a[i]]); //在这之前被计算的权值是无效的,因为之前还不存在这个点; } else if (op[i] == 2) { bit.update(L[pos[i]], a[i]); // 在树状数组上差分; bit.update(R[pos[i]] + 1, -a[i]); } else { int ans = bit.sum(L[pos[i]]) - w[L[pos[i]]]; // 总的权值减去无效的部分; print(ans); puts(""); } } return 0; }