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

题意：

给你一颗树，从根节点出发到叶子节点，如果中间有连续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);
}