Cod sursa(job #2502467)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 30 noiembrie 2019 21:42:42
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.7 kb
#include <bits/stdc++.h>
#define NMAX 16005
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
int n,dp[NMAX],sol;
vector <int> graf[NMAX];
void dfs(int nod,int tata)
{
    for(int i=0;i<(int)graf[nod].size();i++)
    {
        int to=graf[nod][i];
        if(to!=tata)
        {
            dfs(to,nod);
            if(dp[to]>0)
                dp[nod]+=dp[to];
        }
    }
}
int main()
{
    int x,y;
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>dp[i];
    for(int i=1;i<=n-1;i++)
    {
        cin>>x>>y;
        graf[x].push_back(y);
        graf[y].push_back(x);
    }
    sol=-1e9;
    dfs(1,0);
    for(int i=1;i<=n;i++)
        sol=max(sol,dp[i]);
    cout<<sol;
    return 0;
}