Cod sursa(job #688736)

Utilizator dtoniucDaniel Toniuc dtoniuc Data 23 februarie 2012 19:51:26
Problema Asmax Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <iostream>
#include <fstream>
#include <climits>

#define INF INT_MAX/2
 
using namespace std;
short a[16001][16001];
int n,v[16001],viz[16001],m;
ofstream fout("asmax.out");
 
void citire()
{
	ifstream fin("asmax.in");
	fin>>n;
	for(int i=1;i<=n;i++)
		fin>>v[i];
	for(int i=1;i<n;i++)
	{
		int x,y;
		fin>>x>>y;
		a[x][++a[x][0]]=y;
	}
}
int max (int a,int b)
{
	if(a>b) return a;
	return b;
}
int det(int vf)
{
	int sum=-INF;
	for(int i=1;i<=a[vf][0];i++)
		if(viz[a[vf][i]]==0)
		{
			viz[a[vf][i]]=1;
			int p=det(a[vf][i]);
			if(sum>=0 && p>0)
				sum+=p;
			else if(p>sum)
				sum=p;
		}
	v[vf]=max(v[vf],v[vf]+sum);
	return v[vf];
}

int main()
{
	citire();
	viz[1]=1;
	det(1);
	m=-INF;
	for(int i=1;i<=n;i++)
		if(v[i]>m) m=v[i];
	fout<<m;
	return 0;
}