CodeForces - 580C Kefa and Park(树上的 dfs)

CodeForces - 580C Kefa and Park

题目要求,在一课树上开始遍历,不能连续遇到超过m只猫.

树上的dfs:

1建树,用vector

2dfs: 注意根节点是没有父节点的,在写遍历的约束条件的时候要单独考虑根节点. 遍历,能够走到叶子节点就让ans++.

#include <bits/stdc++.h>
#define ll long long
#define maxn 100008
using namespace std;
int n,m;
int a[maxn];
vector<int>ztt[maxn];
int ans;
void dfs(int cur,int miao,int fu,int summ){

    if(summ>m){
        return;
    }
    //cout<<"cur"<<cur<<" "<<"size"<<ztt[cur].size()<<endl;
    if(ztt[cur].size()==1&&cur!=1)
        ans++;
    else{
            for(int i=0;i<ztt[cur].size();i++){
                if(ztt[cur][i]==fu)
                    continue;
                if(a[ ztt[cur][i] ])
                    dfs(ztt[cur][i],a[ztt[cur][i]],cur,summ+1);
                if(!a[ ztt[cur][i] ] )
                    dfs(ztt[cur][i],a[ztt[cur][i]],cur,0);
            }
    }
}
int main() {
    ans=0;
   scanf("%d%d",&n,&m);
   for(int i=1;i<=n;i++){
       scanf("%d",&a[i]);
   }
   for(int i=0;i<n-1;i++){
       int x,y;
       scanf("%d%d",&x,&y);
       ztt[x].push_back(y);
       ztt[y].push_back(x);
   }
   dfs(1,a[1],-1,a[1]);
   printf("%d\n",ans);
   return 0;
}