Codeforces 580C Kefa and Park

题意：

给你一张n个点n-1条边的无向图，1号节点为起点，其它度为1的节点为终点，并给出那几个点有猫，问能到达多少个终点，且路上不能连续经过m只猫。

思路：

先把无向图建成一颗树，然后从1号节点开始往下深搜，看最后能搜到几个叶子节点就是答案了。

#include<cstdio>
#include<vector>
using namespace std;

const int MAX=100005;
int n,m,fa[MAX],a[MAX],ans;
vector<int> gr[MAX];

void DFS1(int u,int f){
fa[u]=f;
int nc=gr[u].size();
for(int i=0;i<nc;i++){
int v=gr[u][i];
if(v!=f) DFS1(v,u);
}
}

void DFS2(int u,int f,int num){
if(num>m) return;
int nc=gr[u].size();
if(nc==1&&gr[u][0]==f){
ans++;
return;
}
for(int i=0;i<nc;i++){
int v=gr[u][i];
if(v!=f){
if(a[v]) DFS2(v,u,num+a[v]);
else DFS2(v,u,a[v]);
}
}
}

void solve(){
ans=0;
DFS2(1,-1,a[1]);
}

int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
gr[i].clear();
}
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
gr[u].push_back(v);
gr[v].push_back(u);
}
DFS1(1,-1);
solve();
printf("%d\n",ans);
return 0;
}