记录编号 543850 评测结果 AAAAAAAAAA
题目名称 没有上司的舞会 最终得分 100
用户昵称 Gravatar代码的搬运工V 是否通过 通过
代码语言 C++ 运行时间 0.019 s
提交时间 2019-10-10 21:09:36 内存使用 13.89 MiB
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=6010;
int n,r;
vector<int> tree[N];
int h[N];
int dp[N][2];
int f[N]; 
void dfs(int u)
{
	dp[u][0]=0;
	dp[u][1]=h[u];
	for(int i=0;i<tree[u].size();i++)
	{
		int v=tree[u][i];
		dfs(v);
		dp[u][1]+=dp[v][0];
		dp[u][0]+=max(dp[v][0],dp[v][1]);
	}
}
int main()
{
	freopen("partyy.in","r",stdin);
	freopen("partyy.out","w",stdout);
	scanf("%d",&n);
	memset(f,-1,sizeof(f));
	for(int i=1;i<=n;i++)
	scanf("%d",&h[i]);
	for(int i=1;i<=n;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		if(a!=0&&b!=0)
		{
			tree[b].push_back(a);
			f[a]=b;	
		}
	}
	int root=1;
	while(f[root]!=-1)
	root=f[root];
	dfs(root);
	int a=max(dp[root][0],dp[root][1]);
	cout<<a;
	return 0;
}