BZOJ 2783: [JLOI2012]树

最新推荐文章于 2020-07-02 17:37:00 发布

cgh_Andy

最新推荐文章于 2020-07-02 17:37:00 发布

阅读量351

点赞数 1

CC 4.0 BY-SA版权

分类专栏： bzoj 贪心

本文链接：https://blog.youkuaiyun.com/cgh_Andy/article/details/77856120

bzoj 同时被 2 个专栏收录

151 篇文章

订阅专栏

贪心

11 篇文章

订阅专栏

本文介绍了一种利用队列解决树形结构中路径和问题的方法。通过递归深度优先搜索，结合滑动窗口思想，实现了O(n)的时间复杂度。文章提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这道题感觉可以随便水吧。。。
不过我觉得我的做法还是挺有趣的（因为辣鸡的我没在树上玩过队列）
搞个队列 >S的时候就减去队头（因为都是正整数）
这样就O（n）了还很短呢~

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
inline int read(){
    int x=0,f=1; char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0' && ch<='9'){x=(x<<1)+(x<<3)+ch-'0'; ch=getchar();}
    return x*f;
}
int nex[N],go[N],fir[N],a[N],s;
int ans=0,q[N];
void dfs(int x,int l,int r,int c){
    while(c>s && l<=r)c-=a[q[l++]];
    if(c==s)ans++;
    for(int k=fir[x];k;k=nex[k]){
        int y=go[k];
        q[r+1]=y; dfs(y,l,r+1,c+a[y]);
    }
}
int main(){
    int n=read(); s=read();
    for(int i=1;i<=n;++i)a[i]=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        nex[i]=fir[x],fir[x]=i,go[i]=y;
    }
    q[1]=1; dfs(1,1,1,a[1]);
    printf("%d\n",ans);
    return 0;
}