CodeForces 580C

这个就是遍历一个树，让你找从头结点到未结点，含标记连续不超多k有多少个。。。
这个题，经典在于一开始我并不知道怎么解决是否为叶子节点的问题，这个题吧，就解决了。。。。

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int MaxN = 1e5;int n , m , ans , a[MaxN + 5];int all , last[MaxN + 5] , pre[2 * MaxN + 5] , other[2*MaxN + 5];void Build(int x , int y){    pre[++all] = last[x];    last[x] = all;    other[all] = y;}void dfs(int x , int fa , int cnt){    if(cnt > m) return;//超过数量就返回    int num = 0;    int ed = last[x];    while(ed != -1){        int dr = other[ed];        if(dr != fa){            if(a[dr] == 0) dfs(dr , x , 0);            else dfs(dr , x , cnt + 1);            num++;//找儿子有几个        }        ed = pre[ed];    }    if(num == 0) ans++;//如果没有儿子的话，那么就是叶子节点}int main(){    all = -1 ;    memset(last , -1 , sizeof(last));    scanf("%d%d", &n , &m);    for(int i = 1 ; i <= n ; i++) scanf("%d",&a[i]);    for(int i = 1 ; i <= n - 1 ; i++){        int u , v;        scanf("%d%d", &u , &v);        Build(u,v);        Build(v,u);    }    int cnt ;    ans = 0;    if(a[1] == 0) cnt = 0;    else cnt = 1;    dfs(1 , 0 , cnt);    printf("%d\n",ans);}