Codeforces 580C Kefa and Park 题解

最新推荐文章于 2020-06-14 11:08:41 发布

DarkoDu

最新推荐文章于 2020-06-14 11:08:41 发布

阅读量341

点赞数

分类专栏： Codeforces 文章标签： codeforces

本文链接：https://blog.csdn.net/m0_37729344/article/details/71430007

版权

Codeforces 专栏收录该内容

62 篇文章 0 订阅

订阅专栏

题意

给定一颗以1为根的树，每个节点有一个权值（0或1），从根分别走向每个叶子节点（一直往下）且每一次至多连续经过m个权值为1的节点，问能成功到达多少叶子节点

思路

dfs时记录一下连续经过权值为1的数量，如果超过就return，否则继续往下，到叶节点计数

代码

#include <cstdio>
#include <vector>
using namespace std;
vector<int> mp[100001];
int a[100001];
int ans;
int n,m;
void dfs(int x,int cnt,int from)
{
    if(a[x]==0)
        cnt=0;
    else cnt++;
    if(cnt>m)
        return;
    if(mp[x].size()==1&&mp[x][0]==from)
    {
        ans++;
        return;
    }
    for(int i=0;i<mp[x].size();i++)
        if(mp[x][i]!=from)
            dfs(mp[x][i],cnt,x);
    return;
}
int main()
{
    int x,y;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=0;i<n-1;i++)
    {
        scanf("%d%d",&x,&y);
        mp[x].push_back(y);
        mp[y].push_back(x);
    }
    dfs(1,0,0);
    printf("%d\n",ans);
    return 0;
}