题意
一个树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();
}
