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

题面

链接

C. Kefa and Park

题意

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

题解

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

总结

这道题目主要是dfs的实现，树的遍历，以及在遍历过程中维护相关信息。同时需要考虑一些细节，特殊情况比如树是一条链。