C. Kefa and Park

题意

一个树n个节点n-1条边，每个节点都有权重，判断从根节点到叶子节点的路径中没有连续权重为1的叶子节点。

题解

普通建图，直接暴力，维护好参数即可。唯一无语的地方就是看英文题，看着看着就忘记注意题目上的细节了，没看到consecutive，连续…

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int a[N];
int n,m;
int h[N],e[N],ne[N],idx;
bool book[N];
int cnt=0;
void add(int u,int v){
    e[idx]=v,ne[idx]=h[u],h[u]=idx++;
}
void dfs(int x,int pr,int cat){
    if(h[x]==-1)cnt+=(cat<=m);
    if(cat>m)return;
    for(int i=h[x];i!=-1;i=ne[i]){
        int j=e[i];
        dfs(j,a[i],(pr==a[j])?cat+1:0);
    }
}
void solve(){
    memset(h,-1,sizeof(h));
    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);
        add(u,v);
    }
    dfs(1,a[1],a[1]);
    printf("%d",cnt);
}
int main() {
    solve();
}