求条,样例输出1
查看原帖
求条,样例输出1
1419569
Z_kazuha楼主2024/10/5 21:25
#include <bits/stdc++.h>
using namespace std;
const int N=6006;
int n,w[N];
bool vis[N];
struct node{
	int nxt,to;
}e[N];
int head[N],cnt,f[N][2];
void add(int u,int v){
	e[++cnt].to=v;
	e[cnt].nxt=head[u];
	head[u]=cnt;
}
void dfs(int u,int father){
	f[u][0]=0;
	f[u][1]=w[u];
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v==father)continue;
		dfs(v,u);
		f[u][0]+=max(f[v][1],f[v][0]);
		f[u][1]+=f[v][0];
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)cin>>w[i];
	for(int i=1;i<n;i++){
		int u,v;
		cin>>u>>v;
		vis[u]=1;
		add(u,v);
	}
	for( int i=1;i<=n;i++){
		if(!vis[i]){
			dfs(i,0);
			cout<<max(f[i][0],f[i][1]);
			return 0;
		}
	}
	return 0;
}
2024/10/5 21:25
加载中...