当前位置: 代码迷 >> 综合 >> Codeforces Round #321 (Div. 2) C. Kefa and Park
  详细解决方案

Codeforces Round #321 (Div. 2) C. Kefa and Park

热度:36   发布时间:2023-12-17 03:36:08.0

题意:

给你一颗树,从根节点出发到叶子节点,如果中间有连续m个标记的点就返回,问有多少个方法

思路:

dfs水题,模拟跑跑就好了

错误及反思:

写的太快,忘了考虑只有一个儿子的情况了。。。

代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 100100;
int n,m;
int arr[N];
int ans=0;
vector<int> v[N];
void dfs(int now,int fa,int ca)
{if(arr[now]) ca++;else ca=0;if(ca>m) return ;if(v[now].size()==1&&now!=fa){ans++;return;}for(int i=0;i<v[now].size();i++){if(v[now][i]!=fa)dfs(v[now][i],now,ca);}return ;
}
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<=n;i++)scanf("%d",&arr[i]);for(int i=1,tempa,tempb;i<n;i++){scanf("%d%d",&tempa,&tempb);v[tempa].push_back(tempb);v[tempb].push_back(tempa);}dfs(1,1,0);printf("%d\n",ans);
}
  相关解决方案