CF#321-C - Kefa and Park-DFS

http://codeforces.com/contest/580/problem/C

题意：

给n，m；

n个数构成一棵树

接下来n个数，表示每个点的权值（1，0） 1表示该节点有猫，0表示没

接下来n-1条边

 找出  有多少个     叶子节点，满足 叶子节点到根节点1的路径上 连续的 权值为1的点不超过 m个

直接dfs遍历邻接表，把节点的权值传递一遍

如果上一个点（累计连续权值）>m,  其下所有儿子都直接记为m+1

如果没，则上一个节点权值为1，【如果当前节点权值也为1，则当前累计权值为上次的+1，如果当前为0，当前累计权值更新为0】

   如果上一个节点权值为0 ，【当前节点累计权值与 自己的权值一样】

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
#include <queue> 
#include <set>
#include <vector>
using namespace std;


int cat[100005]; 
int du[100005];
int tm[100005];
vector <int> map[100005];
int vis[100005];
int m;
int main()
{
	void dfs(int j);
	int n,i; 
	scanf("%d%d",&n,&m);
	int a,b; 
	for (i=1;i<=n;i++)
	{
		scanf("%d",&tm[i]);
	}
	
	for (i=1;i<=n-1;i++)
	{
		scanf("%d%d",&a,&b);
		du[a]++;
		du[b]++; 
		map[a].push_back(b);
		map[b].push_back(a);
	}	 
	
	
	cat[1]=tm[1];
	vis[1]=1;
	dfs(1);
	
	int cun=0;
	for (i=2;i<=n;i++) //注意i=1不能算是答案
	{
		if (du[i]==1&&cat[i]<=m)
			cun++;
	}
	
	
	printf("%d\n",cun);
	return 0;
	
}

void dfs(int j)
{
	int i;
	for (i=0;i<map[j].size();i++)
	{
		
		int t=map[j][i];
		if (vis[t])		//标记已经访问过
			continue;
		if (cat[j]>m)  //如果有一个点处已经超m，那么往下所有点都不合法
		{
			cat[t]=m+1; 
		}
		else
		{
			
			if (tm[j]==1)
			{
				if (tm[t]==1)
					cat[t]=cat[j]+1;
				else
					cat[t]=0;
			}
			else 
				cat[t]=tm[t]; 
		}
		vis[t]=1;
		dfs(t);
	}
	
}