Performance Review Gym - 101174F （DFS + 线段树）

Performance Review Gym - 101174F （DFS + 线段树）

题目链接
题目大意：有一颗有根数，N个节点，每个节点有两个值rank，t，求每个节点的子树中rank值比祖先小的所有结点的t的值的和。
样例：
5
i号节点的根 rank[i] t[i]
4 4 80
1 1 40
-1 10 60
3 5 50
4 8 70
题目分析：
画出图来之后，就发现要跑dfs，然后线段树来维护一些东西，维护什么呢。首先我们要求子树中rank值比祖先小的和，很明显要维护一个和，比如说，当前节点的rank值是5的话，我们去线段树里面查询的话，就要查询1-4的t值。然后你就会发现一个问题，我们要的是子树，但是dfs跑到一个节点的时候，线段树记录的是他的祖先的，怎么办呢，画出图来手动模拟一次dfs过程就会发现，第二个经过该节点时查询的值，和第一次经过该节点时查询的值的差值就是这个节点的子树的我们要求的值。

#include <bits/stdc++.h>
using namespace std;

#define lson root << 1
#define rson root << 1 | 1
#define MID int mid = (left + right) / 2;

const int maxn = 1e5 + 10;
typedef long long ll;

ll tree[maxn << 2], ans1[maxn], ans2[maxn], Rank[maxn], t[maxn];
vector<int> data[maxn];

void updata(int root, int left, int right, int pos, int k) {
    if(pos < left || pos > right) return ;
    if(left == right) {
      tree[root] += k;
       return ;
    }

    MID;
    updata(lson, left, mid, pos, k);
    updata(rson, mid + 1, right, pos, k);
    tree[root] =  tree[lson] + tree[rson];
}

ll query(int root, int left, int right, int ql, int qr) {
    if(qr < left || ql > right) return 0;
    if(ql <= left && right <= qr) return tree[root];

    MID;
    return query(lson, left, mid, ql, qr) + query(rson, mid + 1, right, ql, qr);
}

void DFS(int s) {
  // printf("s = %d\n", s);
   ans1[s] = query(1, 0, maxn, 0, Rank[s] - 1);
   //printf("ans1[%d] = %d\n", s, ans1[s]);
   updata(1, 0, maxn, Rank[s], t[s]);
   for(int i = 0; i < data[s].size(); i++) {
       DFS(data[s][i]);
       ans2[data[s][i]] = query(1, 0, maxn, 0, Rank[data[s][i]] - 1);
    //   printf("ans2[%d] = %d\n", data[s][i], ans2[data[s][i]]);
   }
}

int main()
{
    int n, top;
    scanf("%d", &n);
    memset(tree, 0, sizeof(tree));
    for(int i = 1; i <= n; i++)
    {
        int u;
        scanf("%d %lld %lld", &u, &Rank[i], &t[i]);
        if(u == -1) {
            top = i;
        }
        else data[u].push_back(i);
    }

    DFS(top);
    ans2[top] = query(1, 0, maxn, 0, Rank[top] - 1);

    for(int i = 1; i <= n; i++)
    {
        printf("%lld\n", ans2[i] - ans1[i]);
    }
    return 0;
}