Cod sursa(job #3141249)

Utilizator amunnumeVlad Patrascu amunnume Data 13 iulie 2023 13:36:20
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <bits/stdc++.h>
#define N 16005
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int n,i,a,b,mx=-999999999,v[N],s[N];
vector<int> q[N];
void dfs(int p,int x)
{
    int i,y;
    s[x]=v[x];
    for(i=0;i<q[x].size();++i)
    {
       y=q[x][i];
       if(y!=p) {dfs(x,y);
       if(s[y]>0) s[x]+=s[y];}
    }
}
int main()
{
    fin>>n;
    for(i=1;i<=n;++i)
    {
        fin>>v[i];
    }
    for(i=1;i<n;++i)
    {
        fin>>a>>b;
        q[a].push_back(b);
        q[b].push_back(a);
    }
    dfs(0,1);
    for(i=1;i<=n;++i)
    {
        mx=max(mx,s[i]);
    }
    fout<<mx;
    return 0;
}