Cod sursa(job #2873474)

Utilizator bostanlucastefanBostan Luca-Stefan bostanlucastefan Data 19 martie 2022 10:39:26
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <fstream>
#include <vector>
#define pb push_back

using namespace std;
using vi=vector<int>;

ifstream fin("asmax.in");
ofstream fout("asmax.out");

const int N=16e3+2;
const int INF=-1e9;

int n,u,v,i;
int ans=INF;
int dp[N];
vi g[N];

void dfs(int x,int p)
{
    int sum=0;
    for(auto i:g[x])
        if(i!=p)
        {
            dfs(i,x);
            if(dp[i]>0)
                sum+=dp[i];
        }
    dp[x]+=sum;
}

int main()
{
    fin>>n;
    for(i=1; i<=n; i++)
        fin>>dp[i];

    for(i=1; i<n; i++)
    {
        fin>>u>>v;
        g[u].pb(v);
        g[v].pb(u);
    }

    dfs(1,0);
    for(i=1; i<=n; i++)
        ans=max(ans,dp[i]);
    fout<<ans<<'\n';
    return 0;
}