Cod sursa(job #536108)

Utilizator cosmin79Carabet Cosmin Andrei cosmin79 Data 18 februarie 2011 11:37:09
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <fstream>
#include <vector>
#define NMAX 16005
#define INF 2000000000
#define pb push_back
using namespace std;
int n,A[NMAX],S[NMAX],rez=-INF;
vector <int> B[NMAX];
char viz[NMAX];
ifstream in("asmax.in");
ofstream out("asmax.out");
void read()
{
	in>>n;
	int i,x,y;
	for (i=1; i<=n; i++)
		in>>A[i];
	for (i=1; i<n; i++)
	{
		in>>x>>y;
		B[x].pb(y); B[y].pb(x);
	}
}
void dfs(int nod)
{
	viz[nod]=1; S[nod]=A[nod];
	int i,vec;
	for (i=0; i<B[nod].size(); i++)
	{
		vec=B[nod][i];
		if (!viz[vec])
		{
			dfs(vec);
			if (S[nod]<S[nod]+S[vec])
				S[nod]=S[nod]+S[vec];
		}
	}
	if (S[nod]>rez) 
		rez=S[nod];
}
int main()
{
	read();
	dfs(1);
	out<<rez<<'\n';
	return 0;
}