Cod sursa(job #688652)

Utilizator dtoniucDaniel Toniuc dtoniuc Data 23 februarie 2012 18:44:11
Problema Asmax Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.98 kb
#include <iostream>
#include <fstream>
#include <climits>

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

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