记录编号 543849 评测结果 AAAAAAAAAA
题目名称 没有上司的舞会 最终得分 100
用户昵称 Gravatar斯内普和骑士 是否通过 通过
代码语言 C++ 运行时间 0.011 s
提交时间 2019-10-10 21:08:54 内存使用 13.89 MiB
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int maxn=6001;
int n;
int father[maxn];
int dp[maxn][2];//dp中0代表不取,1代表取 
int happy[maxn];
vector<int> G[maxn];
void dfs(int u)
{
	dp[u][0]=0;
	dp[u][1]=happy[u];
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		dfs(v);
		dp[u][0]+=max(dp[v][0],dp[v][1]);
		dp[u][1]+=dp[v][0];
	}
}
int main()
{
	freopen("partyy.in","r",stdin);
	freopen("partyy.out","w",stdout);
	memset(father,-1,sizeof(father));
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	scanf("%d",&happy[i]);
	while(true)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		if(x==0&&y==0)
		break;
		G[y].push_back(x);
		father[x]=y;
	}
	int root=1;
	while(father[root]!=-1)
	root=father[root];
	dfs(root);
	printf("%d",max(dp[root][0],dp[root][1]));
	return 0;
}