dfs满江红求条

#include<bits/stdc++.h>
using namespace std;
int n,k,color[1005];
vector<int> g[1005];
int maxn=-1e9;
bool flag[1005];
void dfs(int cnt,int ans,int fa){
    if(cnt>k){
        return;
    }
    maxn=max(maxn,ans);
    for(int j=0;j<g[fa].size();j++){
    	if(flag[g[fa][j]]==0){
	        if(color[g[fa][j]]){
	            flag[g[fa][j]]=1;
	            dfs(cnt+1,ans+1,g[fa][j]);
	            flag[g[fa][j]]=0;
	        }
	        else{
	            flag[g[fa][j]]=1;
	            dfs(cnt,ans+1,g[fa][j]);
	            flag[g[fa][j]]=0;
	        }
	    }
	}
}
int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>color[i];
    }
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    for(int i=1;i<=n;i++){
        if(color[i]==0){
            dfs(0,1,i);
        }
        else{
            dfs(1,1,i);
        }
    }
    cout<<maxn;
    return 0;
}