Cod sursa(job #36818)

Utilizator IgnitionMihai Moraru Ignition Data 24 martie 2007 09:05:13
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <stdio.h>
#include <limits.h>
#include <vector>

using namespace std;

#define MN (16384)

int N, n[MN], v[MN], X = LONG_MIN;
vector<int> G[MN];

int c(int node)
{
	int i;
	v[node] = 1;
	for(i = 0; i < G[node].size(); ++i) if(!v[G[node][i]] && c(G[node][i]) > 0)
		n[node] += n[G[node][i]];
	if(n[node] > X)
		X = n[node];
	return n[node];
}

int main()
{
	int i;

	freopen("asmax.in", "r", stdin);
	freopen("asmax.out", "w", stdout);

	scanf("%d", &N);
	for(i = 0; i < N; ++i)
		scanf("%d", &n[i]);
	for(i = 0; i < N-1; ++i) {
		int n1, n2;
		scanf("%d %d", &n1, &n2); --n1; --n2;
		G[n1].push_back(n2);
		G[n2].push_back(n1);
	}

	c(0);

	printf("%d\n", X);

	return 0;
}