叶子节点(图的遍历)

Kingcarry6

已于 2023-04-05 14:12:20 修改

阅读量124

点赞数

分类专栏：图论文章标签：数据结构算法

于 2023-04-01 21:12:08 首次发布

本文链接：https://blog.csdn.net/Kingcarry6/article/details/129903147

版权

图论专栏收录该内容

13 篇文章 0 订阅

订阅专栏

4946. 叶子节点 - AcWing题库

输入

输出

输入

7 1
1 0 1 1 0 0 0
1 2
1 3
2 4
2 5
3 6
3 7

输出

#include <iostream>
#include <vector>
#include <bits/stdc++.h>
#include <unordered_map>
#include <queue>
#include <algorithm>
#define x first
#define y second
#define pb emplace_back
#define fu(i,a,b) for(int i=a;i<=b; ++ i)
#define fd(i,a,b) for(int i=a;i>=b;	-- i)
#define endl '\n'
#define ms(x,y) memset(x,y,sizeof x)
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;


typedef long long LL;
typedef unsigned long long ULL;
typedef vector<vector<LL>> VVL;
typedef vector<vector<int>> VVI;
typedef vector<LL> VL;
typedef vector<bool> VB;
typedef vector<int> VI;
typedef vector<string> VS;
typedef pair<int,int> PII;
typedef vector<PII> VPII;
typedef pair<PII,int> PIII;
typedef pair<double,double> PDD;
typedef pair<double,int> PDI;
typedef pair<char,int> PCI;
typedef pair<string,int> PSI;
typedef pair<int,string> PIS;
typedef pair<int,char> PIC;
typedef pair<LL,LL> PLL;
typedef __int128 i128;
typedef unsigned long long ULL;
const int N =1e5+ 10,M = N * 2  ,INF = 0x3f3f3f3f,P = 131;
const double eps = 1e-8,DNF = 1e18;
const int mod = 998244353,base= 20010;
const LL LNF=(LL) INF * INF;


int n,m;
int c[N],ans;
int e[M],h[N],ne[M],idx;


void add(int a,int b)
{
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
// 没有子节点的 结点 叫叶子结点 
// f -> 这条路径是否存在超过 m个黑色节点连续的排列在一起 
// 1 -> 不存在  0-> 存在 
int dfs(int u,int fa,int cnt,bool f)
{
	if(c[u]) 
	{
		if(++ cnt > m ) f =0 ;
	}
	else cnt =0 ;
	
	int res= 0,son = 0;	
	
	for(int i=h[u];~i;i=ne[i])
	{
		int j= e[i];if(j == fa) continue;
		son ++ ;
		
		res += dfs(j,u,cnt,f);
	}
	// 是叶子结点 且 走到当前是合法的 
	if(!son && f ) res ++ ; 
	return res ;
}



inline void solve()
{
	cin >> n >> m ;
	fu(i,1,n) cin >> c[i];
	ms(h,-1);
	
	fu(i,1,n-1)
	{
		int a,b;cin >> a >> b;
		add(a,b);
		add(b,a);
	}
	cout << dfs(1,-1,0,1) << endl;
	
		
	
}

signed main()
{
//  freopen("1.txt","r",stdin);
//	#define int long long 
//	init(N-1);
    // ios
//	cout << fixed<<setprecision(2);
    int t=1;
//    cin>>t;
    int now = 1;
    while(t -- )
    {
//      cout<<"Case "; 
//      cout<<"Case #"; 
//      cout<<"Scenario #"; 
//      cout<< now ++ <<": ";
//      cout<< now ++ <<": \n";
        solve();
    }


    return 0;
}