比赛 2009noip模拟试卷 评测结果 AAAAAAAAAA
题目名称 火车站饭店 最终得分 100
用户昵称 iortheir 运行时间 0.482 s
代码语言 C++ 内存使用 7.86 MiB
提交时间 2016-10-09 16:22:00
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>

#define pb push_back

using namespace std;

const int maxn = 100000 + 100;
const int maxm = 50 + 5;

int n;
int A[maxn];

vector<int >C[maxn];

bool vis[maxn][maxm];

int a;
int b;

int f[maxn][2];

int cnt1;
int cnt2;

inline void find(int u)
{
	cnt1 = 0;
	cnt2 = 0;
	f[u][1] = A[u]; 
	for(int i=0;i<C[u].size();++i)
	{
		if(!vis[u][i])
		{
			vis[u][i] = true;
			int t = C[u][i];
			for(int j=0;j<C[t].size();++j)
			{
				if(C[t][j]==u)
				{
					vis[t][j] = true;
					break;
				}
			}
			find(t);
			f[u][0]+=max(f[t][1],f[t][0]);
			f[u][1]+=f[t][0];
		}
	}
}

int main()
{
	freopen("profitz.in","r",stdin);
	freopen("profitz.out","w",stdout);
	memset(vis,false,sizeof(vis));
	memset(f,0,sizeof(f));
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
	{
		scanf("%d",&A[i]);
	}
	for(int i=1;i<=n-1;++i)
	{
		scanf("%d%d",&a,&b);
		C[a].pb(b);
		C[b].pb(a);
	}
	find(1);
	cout<<max(f[1][0],f[1][1]);
	return 0;
}