codeforces 580C(Kefa and Park, 搜索类型题)

题目大意：

       输入n,m;  表示有n个点，如果连续遇到m只猫则男主角不能继续前进； 首先输入n个值a1, a2, ...an 如果是1表示有猫，否则没有猫；（这些点会组成树，树的叶子表示餐馆），问能够到达多少餐馆。

题目分析：

    从树根出发，多增加一个参数cur表示连续遇到的猫的个数，如果超过k，则不能继续往下走。 记录能够到达的叶子;

这道题需要注意的是，1 .边是双向联通的，搜索时应该保存一个数据pre表示之前的点，已经走的点不能够再走回去；2  根节点的度数也可能是1，如果更具度数判断，要注意根节点的判断；

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define F first
#define S second

using namespace std;

vector<int> iv[100000+10];
int n, m, tot;
int isCat[100000+10];

void dfs(int u, int pre, int cur){   //pre表示之前走过的点， 不应该再走;
    if(iv[u].size() == 1 && cur+isCat[u] <= m && pre != -1){  // 注意,根节点的度数可能为1;
        tot++;
    } else{
        if(isCat[u]) ++cur;
        else cur = 0;
        if(cur > m) return ;
        for(unsigned i = 0; i < iv[u].size(); ++i){
            int v = iv[u][i];
            if(v != pre)
                dfs(v, u, cur);
        }
    }
}

int main()
{
    while(scanf("%d%d", &n, &m) != EOF){
        for(int i = 1; i <= n; ++i){
            scanf("%d", &isCat[i]);
        }
        for(int i = 0; i < n-1; ++i){
            int u, v;
            scanf("%d%d", &u, &v);
            iv[u].push_back(v);
            iv[v].push_back(u);
        }
        tot = 0;
        dfs(1, -1, 0);
        printf("%d\n", tot);
    }
    return 0;
}