Cod sursa(job #570035)

Utilizator swift90Ionut Bogdanescu swift90 Data 2 aprilie 2011 13:47:02
Problema Asmax Scor 100
Compilator cpp Status done
Runda preselectie_acm_unibuc Marime 0.95 kb
#include<stdio.h>
#include<vector>
using namespace std;
vector<int> nr[16100];
int N,SOL;
int val[16100],sol[16100];
char viz[16100];
void df(int x){
	viz[x]=1;
	for(vector<int>::iterator it=nr[x].begin();it!=nr[x].end();++it){
		if(!viz[*it])
			df(*it);
	}
	int su=0;
	for(vector<int>::iterator it=nr[x].begin();it!=nr[x].end();++it){
		if(viz[*it]!=2)
			continue;
		if(sol[*it]>0)
			su+=sol[*it];
	}
	sol[x]=val[x]+su;
	viz[x]=2;
}
int main(){
	freopen("asmax.in","r",stdin);
	freopen("asmax.out","w",stdout);
	int x,y,i,neg=-100000;
	scanf("%d",&N);
	for(i=1;i<=N;++i){
		scanf("%d",&val[i]);
		if(val[i]>neg)
			neg=val[i];
	}
	if(neg<=0){
		printf("%d\n",neg);
		return 0;
	}
	for(i=1;i<N;++i){
		scanf("%d%d",&x,&y);
		nr[x].push_back(y);
		nr[y].push_back(x);
	}
	df(1);
	SOL=sol[1];
	for(i=1;i<=N;++i){
		if(SOL<sol[i])
			SOL=sol[i];
	}
	printf("%d\n",SOL);
	fclose(stdin);
	fclose(stdout);
	return 0;
}