Codeforces Round #321 (Div. 2) C. Kefa and Park time limit per test

题目大意：一棵树，有的节点上有猫，有的没有。求出从根节点出发，到达叶子节点的路径中连续出现猫的数量不能超过m的路径有几条。

解题思路：DFS。细节：vector构造树时因为不知道谁是父亲节点谁是子节点，因此两个点都当做父亲节点处理。判断叶子节点时，儿子个数为1的就是叶子节点。

#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
const int maxm=1e5+10;
int vis[maxm];
int a[maxm];
vector<int>v[maxm];
int sum;
int n,m;
void dfs(int x,int cnt)
{
if(cnt>m)
{
return;
}
int len=v[x].size();
if(len==1&&x!=1)
{
if(a[x])
{
cnt++;
}
if(cnt>m)
{
return;
}
sum++;
return;
}
for(int i=0;i<len;i++)
{
if(vis[v[x][i]])
{
continue;
}
else
{
vis[v[x][i]]=1;
}
if(a[x])
{
dfs(v[x][i],cnt+1);
}
else
{
dfs(v[x][i],0);
}
}
return;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
int x,y;
memset(vis,0,sizeof(vis));
vis[1]=1;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
for(int i=0;i<n-1;i++)
{
scanf("%d%d",&x,&y);
v[x].push_back(y);
v[y].push_back(x);
}
sum=0;
dfs(1,0);
printf("%d\n",sum);
}
return 0;
}