比赛 20200109 评测结果 AAAAAAAAAA
题目名称 猴腮雷 最终得分 100
用户昵称 fmq03 运行时间 0.063 s
代码语言 C++ 内存使用 13.87 MiB
提交时间 2019-12-25 19:42:49
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<vector>
#include<cmath>
using namespace std;
int fun[6001];
int n;
vector<int>son[6001];
int f[6001][2];
bool vis[6001];
void dfs(int u){
	vis[u]=true;
	for(int i=0;i<son[u].size();++i){
		if(!vis[son[u][i]]){
			dfs(son[u][i]);
			f[u][1]+=f[son[u][i]][0];
			f[u][0]+=max(f[son[u][i]][1],f[son[u][i]][0]);
		}
	}
	return;
}
int main(){
	freopen("monk.in","r",stdin);
	freopen("monk.out","w",stdout);
	cin>>n;
	for(int i=1;i<=n;++i){
		cin>>fun[i];
		f[i][1]=fun[i];
	}
	for(int i=1;i<n;++i){
		int a,b;
		cin>>a>>b;
		son[a].push_back(b);
		son[b].push_back(a);
	}
	dfs(1);
	cout<<max(f[1][0],f[1][1])<<endl;
	return 0;
}