luoguP2982 [USACO10FEB]慢下来Slowing down

https://www.luogu.org/problemnew/show/P2982

这题你写个树剖当然可以做，但是我们还有一种更简单的方法，使用 dfs 序 + 树状数组即可

考虑一只牛到了自己的地方后会对哪些牛产生贡献

当然是它的子树中的牛啊

所以维护一下每个点的 size 和 dfs 序，树状数组维护差分数组，单点查询就转换为前缀查询

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

const int N = 1e5 + 5;

struct Edge {
    int u, v, next;
}G[N << 1];

int head[N], tops[N], siz[N], f[N];
int n, cnt, tot;

inline void addedge(int u, int v) {
    G[++tot] = (Edge) {u, v, head[u]}, head[u] = tot;
    G[++tot] = (Edge) {v, u, head[v]}, head[v] = tot;
}

void dfs(int u, int fa) {
    tops[u] = ++cnt, siz[u] = 1;
    for(int i = head[u]; i; i = G[i].next) {
        int v = G[i].v;
        if(v == fa) continue;
        dfs(v, u);
        siz[u] += siz[v];
    }
}

int lowbit(int x) {return x & -x;}
void add(int x, int y) {for(int i = x; i <= n; i += lowbit(i)) f[i] += y;}
int query(int x) {int ans = 0; for(int i = x; i; i -= lowbit(i)) ans += f[i]; return ans;}

int main() {
    cin >> n;
    for(int i = 1; i < n; i++) {
        int a, b;
        scanf("%d %d", &a, &b);
        addedge(a, b);
    }
    dfs(1, 0);
    for(int i = 1; i <= n; i++) {
        int a; scanf("%d", &a);
        printf("%d\n", query(tops[a]));
        add(tops[a], 1);
        add(tops[a] + siz[a], -1);
    }
    return 0;
}