Cod sursa(job #2502468)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 30 noiembrie 2019 21:43:09
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.69 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;
    f>>n;
    for(int i=1;i<=n;i++)
        f>>dp[i];
    for(int i=1;i<=n-1;i++)
    {
        f>>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]);
    g<<sol;
    return 0;
}