Codeforce-580C(Kefa and Park)

#include <bits/stdc++.h>
using namespace std;
int n,m;
vector<int>v[100010];
int visit[100010];
bool flag[100010];
int cnt=0;
//就是一个树的dfs,其实蛮简单的，菜鸡瑟瑟发抖
void dfs(int i,int con,int maxcon)
{
    visit[i]=1;
    if(v[i].size()==1&&i!=1)//这个地方判断是否是叶子，就是v[i].size()==1但是根的size()也可能为1
    {
        if(maxcon<=m)
        {
            cnt++;
        }
    }
    for(int j=0;j<v[i].size();++j)
    {
        if(!visit[v[i][j]])
        {
            if(flag[v[i][j]]==1)
            {
                dfs(v[i][j],con+1,max(maxcon,con+1));//这个地方出了bug，
            }//像我这种写法其实在当前递归函数了已经开始考虑下一步了，所以递归结束的条件直接判断即可
            else//不用再对最后一个叶子节点单独判断
            {
                dfs(v[i][j],0,maxcon);
            }

        }
    }
}
int main(void)
{
    ios::sync_with_stdio(0);
    cin>>n>>m;
    int temp;
    for(int i=1;i<=n;++i)
    {
        cin>>temp;
        if(temp==1)
        {
            flag[i]=true;
        }
        else
        {
            flag[i]=false;
        }
    }
    int a,b;
    for(int i=1;i<=n-1;++i)
    {
        cin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs(1,flag[1],flag[1]);
    cout <<cnt <<endl;
    return 0;
}