BZOJ 2783 树 - 树上倍增 + 二分

传送门

分析:

对每个点都进行一次二分:将该点作为链的底端,二分链顶端所在的深度,然后倍增找到此点,通过前缀和相减求出链的权值,并更新l,r。

code

#include<bits/stdc++.h>
using namespace std;
namespace IO {
    template<typename T>
    inline void read(T &x) {
        T i = 0, f = 1;
        char ch = getchar();
        for(; (ch < '0' || ch > '9') && ch != '-'; ch = getchar());
        if(ch == '-') f = -1, ch = getchar();
        for(; ch >= '0' && ch <= '9'; ch = getchar()) i = (i << 3) + (i << 1) + (ch - '0');
        x = i * f;
    }
    template<typename T>
    inline void wr(T x) {
        if(x < 0) putchar('-'), x = -x;
        if(x > 9) wr(x / 10);
        putchar(x % 10 + '0');
    }
} using namespace IO;

const int N = 1e5 + 5;
int n, s;
vector<int> G[N];
int dep[N], fa[N][25], val[N], sum[N], ans;

inline void dfs(int u, int f) {
    fa[u][0] = f;
    dep[u] = dep[f] + 1;
    sum[u] = val[u] + sum[f];
    for(int i = 1; i <= 20; i++) fa[u][i] = fa[fa[u][i - 1]][i - 1];
    for(int e = G[u].size() - 1; e >= 0; e--) {
        int v = G[u][e];
        if(v == f) continue;
        dfs(v, u);
    }
}

inline int calc(int k, int delta) {
    int p = k;
    for(int i = 20; i >= 0; i--)
        if(delta & (1 << i)) k = fa[k][i];
    return sum[p] - sum[fa[k][0]];
}

inline void solve(int k) {
    int l = 0, r = dep[k];
    while(l <= r) {
        int mid = l + r >> 1;
        int ret = calc(k, dep[k] - mid);
//      cout<<k<<" "<<l<<" "<<r<<" "<<mid<<" "<<ret<<" "<<endl;
        if(ret == s) {
            ans++;
            return;
        } else if(ret < s) r = mid - 1;
        else l = mid + 1;
    }
}

int main() {
    freopen("h.in", "r" ,stdin);
    read(n), read(s);
    for(int i = 1; i <= n; i++) read(val[i]);
    for(int i = 1; i < n; i++) {
        int x, y;
        read(x), read(y);
        G[x].push_back(y), G[y].push_back(x);
    }
    dep[0] = -1, dfs(1, 0);
    for(int i = 1; i <= n; i++) solve(i);
    wr(ans);
    return 0;
}

转载于:https://www.cnblogs.com/CzYoL/p/7767761.html

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值