Cod sursa(job #658053)

Utilizator BarracudaFMI-Alex Dobrin Barracuda Data 7 ianuarie 2012 20:47:59
Problema Asmax Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.65 kb
#include<cstdio>
#include<vector>
#define lim 16001
using namespace std;
int V[lim],n,x,y,maxu,viz[lim],suma[lim],i;
vector<int>G[lim];
void dfs(int nod){
	viz[nod]=1;
	int v;
	
	for(int i=0;i<G[nod].size();++i){
		v=G[nod][i];
		if(!viz[v] ){
			dfs(v);
			if(V[v]>0)
				suma[nod]+=V[v];
		}
	}
	if(suma[nod]+V[nod]>maxu)
		maxu=suma[nod]+V[nod];
}
int main (){
	freopen("asmax.in","r",stdin);
	freopen("asmax.out","w",stdout);
	scanf("%d",&n);
	maxu=-10000;
	for(i=1;i<=n;i++)
		scanf("%d",&V[i]);
	for(i=1;i<=n;i++){
		scanf("%d%d",&x,&y);
		G[x].push_back(y);
		G[y].push_back(x);
	}
	dfs(1);
	printf("%d",maxu);
	return 0;
	
}