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;
}

转载于:https://www.cnblogs.com/LJC00118/p/9613384.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值