记录编号 578012 评测结果 AAAAAAAAAA
题目名称 没有上司的舞会 最终得分 100
用户昵称 Gravatar瑆の時間~無盡輪迴·林蔭 是否通过 通过
代码语言 C++ 运行时间 0.000 s
提交时间 2023-01-12 00:45:18 内存使用 0.00 MiB
显示代码纯文本
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<vector>
#include<map>
using namespace std;
vector<int> b[6005];
map<int, bool> mp;
int val[6005];
int dp[6005][2];
int n,x,y,root;
void DFS(int x)
{
	dp[x][0] = 0;
	dp[x][1] = val[x];
	for (int i = 0; i < b[x].size(); i++)
	{
		int to = b[x][i];
		DFS(to);
		dp[x][1] += dp[to][0];
		dp[x][0] += max(dp[to][0], dp[to][1]);
	}
}
int main()
{

	freopen("partyy.in", "r", stdin);
	freopen("partyy.out", "w", stdout);
	scanf("%d", &n);
	for (int i = 1; i <= n; i++)
		scanf("%d", val + i);
	for (int i = 1; i < n; i++)
	{
		scanf("%d%d", &x, &y);
		b[y].push_back(x);
		mp[x] = 1;
	}
	for (int i = 1; i <= n; i++)
	{
		if (mp.find(i) == mp.end())
		{
			root = i;
			break;
		}
	}
	DFS(root);
	printf("%d", max(dp[root][0], dp[root][1]));
	return 0;
}