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()==&&cnt!=) ans++;
     vis[cnt]++;
     for(int i=; i<tree[cnt].size(); i++)
     {
         if(!vis[tree[cnt][i]])
         {
             if(!a[tree[cnt][i]]) dfs(tree[cnt][i], );
             else dfs(tree[cnt][i], num+);
         }
     }
 }

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