题意描述
给一棵n个节点树，某个结点可能会打上标记，从根节点走，问有几条路径上的连续标记树不超过m。
思路
按照题意dfs即可
AC代码
代码语言：javascript复制#include<bits/stdc .h>
#define x first
#define y second
#define pb push_back
#define IOS ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;
typedef pair<char,char> PCC;
typedef long long LL;
const int N=5*1e5 10;
const int M=150;
const int INF=0x3f3f3f3f;
const int MOD=998244353;
int cat[N];
vector<int> g[N];
int ans,n,m;
void dfs(int u,int cnt,int father){
if(cat[u]) cnt ;
else cnt=0;
if(cnt>m) return;
if(g[u].size()==1 && u!=1){
ans ;
return;
}
for(int i=0;i<g[u].size();i ){
if(g[u][i]==father) continue;
dfs(g[u][i],cnt,u);
}
}
void solve(){
cin>>n>>m;
for(int i=1;i<=n;i ) cin>>cat[i];
for(int i=0;i<n-1;i ){
int x,y;cin>>x>>y;
g[x].push_back(y);
g[y].push_back(x);
}
dfs(1,0,-1);
cout<<ans<<endl;
}
int main(){
IOS;
solve();
return 0;
}
