Pagini recente » Cod sursa (job #1558411) | Cod sursa (job #1413698) | Cod sursa (job #947670) | Cod sursa (job #1845655) | Cod sursa (job #785657)
Cod sursa(job #785657)
#include <stdio.h>
#include <vector>
#include <algorithm>
#define NMax 16010
using namespace std;
const char IN[]="asmax.in",OUT[]="asmax.out";
int N;
int v[NMax] , T[NMax], D[NMax];
vector<int> ad[NMax];
void dfs(int x,int p=0)
{
T[x]=v[x];
for (int i=0;i<(int)ad[x].size();++i) if (ad[x][i]!=p)
dfs(ad[x][i],x),
T[x]+=max(0,T[ad[x][i]]);
}
void dfs2(int x,int p=0)
{
D[x]=max(0,D[p]) + max(0,T[p]) - max(0,T[x]);
for (int i=0;i<(int)ad[x].size();++i) if (ad[x][i]!=p)
dfs2(ad[x][i],x);
}
int main()
{
int i,x,y;
freopen(IN,"r",stdin);
scanf("%d",&N);
for (i=1;i<=N;++i) scanf("%d",v+i);
for (i=1;i<N;++i)
scanf("%d%d",&x,&y),
ad[x].push_back(y),
ad[y].push_back(x);
dfs(1);dfs2(1);
int Rez=D[1]+T[1];
for (i=1;i<=N;++i) if (D[i]+T[i]>Rez)
Rez=D[i]+T[i];
freopen(OUT,"w",stdout);
printf("%d\n",Rez);
fclose(stdout);
return 0;
}