前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >codeforces 580C(dfs)

codeforces 580C(dfs)

作者头像
dejavu1zz
发布2020-10-23 15:19:55
3260
发布2020-10-23 15:19:55
举报

题意描述

给一棵n个节点树,某个结点可能会打上标记,从根节点走,问有几条路径上的连续标记树不超过m。

思路

按照题意dfs即可

AC代码

代码语言:javascript
复制
#include<bits/stdc++.h>
#define x first
#define y second
#define pb push_back
#define IOS ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;
typedef pair<char,char> PCC;
typedef long long LL;
const int N=5*1e5+10;
const int M=150;
const int INF=0x3f3f3f3f;
const int MOD=998244353;
int cat[N];
vector<int> g[N];
int ans,n,m;
void dfs(int u,int cnt,int father){
    if(cat[u]) cnt++;
    else cnt=0;
    if(cnt>m) return;
    if(g[u].size()==1 && u!=1){
        ans++;
        return;
    }
    for(int i=0;i<g[u].size();i++){
        if(g[u][i]==father) continue;
        dfs(g[u][i],cnt,u);
    }

}
void solve(){
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>cat[i];
    for(int i=0;i<n-1;i++){
        int x,y;cin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1,0,-1);
    cout<<ans<<endl;
}
int main(){
    IOS;
    solve();
    return 0;
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-06-14 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 题意描述
  • 思路
  • AC代码
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档