Codeforces Round 303 (Div. 2)C. Kefa and Park（DFS、实现）

本文主要是介绍Codeforces Round 303 (Div. 2)C. Kefa and Park（DFS、实现），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

文章目录

题面
链接
题意
题解
代码
总结

题面

链接

C. Kefa and Park

题意

求叶节点数量，叶节点满足，从根节点到叶节点的路径上最长连续1的长度小于m

题解

这道题目主要是实现，当不满足条件时直接返回。
到达叶节点后统计答案，用vector存图的话，无向图时，叶节点的边只有一条，也就是 $g [i] . s i ze () == 1$ 而不是0
需要特判是一条链的情况，一条链的话根节点的 $g [i] . s i ze () == 1$ 也成立

代码

#include <bits/stdc++.h> 
#define int long long
#define rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(int i = (a); i >= (b); --i)
#define pii pair<int, int>
#define pll pair<long long, long long>
#define ll long long
#define db double
#define endl '\n'
#define x first
#define y second
#define pb push_backusing namespace std;
const int N=1e5+10;
vector<int>g[N];
int a[N],ans,n,m;void dfs(int u,int fa,int sum,int maxx){if(maxx>m){	return;}//统计答案if(g[u].size()==1&&max(maxx,sum+a[u])<=m&&u!=1){
//		cout<<"----------"<<u<<endl;ans++;return;}for(auto y:g[u]){if(y==fa)	continue;if(a[u]==1){if(a[fa]==1){dfs(y,u,sum+1,max(maxx,sum+1));}else{dfs(y,u,1,max(maxx,1*1ll));}}else{dfs(y,u,0,maxx);}}
}void solve()
{cin>>n>>m;rep(i,1,n){cin>>a[i];}rep(i,1,n-1){int u,v;cin>>v>>u;g[u].pb(v);g[v].pb(u);}//当前结点、根节点，目前连续猫数。dfs(1,0,0,0);cout<<ans<<endl;
}signed main(){ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
//   	freopen("1.in", "r", stdin);int _;
//	cin>>_;
//	while(_--)solve();return 0;
}