Codeforces Round #321 (Div. 2)C(tree dfs)

题意：给出一棵树，共有n个节点，其中根节点是Kefa的家，叶子是restaurant，a[i]....a[n]表示i节点是否有猫，问：Kefa要去restaurant并且不能连续经过m个有猫的节点有多少条路径；

思路：先用vector数组建树； 再dfs..（第一次用vector建树，还看了别人的代码，果然是zz)；

代码：

#include <bits/stdc++.h>
#define ll long long
#define MAXN 100000+10
using namespace std;

vector<int>tree[MAXN];
int a[MAXN], vis[MAXN], n, k;
ll ans;

void dfs(int cnt, int num)
{
    if(vis[cnt]||num>k) return;
    if(tree[cnt].size()==1&&cnt!=1) ans++;
    vis[cnt]++;
    for(int i=0; i<tree[cnt].size(); i++)
    {
        if(!vis[tree[cnt][i]])
        {
            if(!a[tree[cnt][i]]) dfs(tree[cnt][i], 0);
            else dfs(tree[cnt][i], num+1);
        }
    }
}

int main(void)
{
    memset(vis, 0, sizeof(vis));
    cin >> n >> k;
    for(int i=1; i<=n; i++)
    {
        tree[i].clear();
        cin >> a[i];
    }
    for(int i=1; i<n; i++)
    {
        int x, y;
        cin >> x >> y;
        tree[x].push_back(y);
        tree[y].push_back(x);
    }
    ans=0;
    dfs(1, a[1]);
    cout << ans << endl;
    return 0;
}