Cod sursa(job #3281011)

Utilizator apoputoaievladVlad Cristian Apoputoaie apoputoaievlad Data 28 februarie 2025 08:32:35
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.87 kb
#include <bits/stdc++.h>

using namespace std;

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

vector <int> l[16002];
int n,viz[16002],costrad[16002],cost[16002];

void Dfs(int nod)
{
    viz[nod]=1;

    for(auto e:l[nod])
        if(viz[e]==0)
        {
        costrad[e]=cost[e];
        Dfs(e);
        if(costrad[nod]+costrad[e]>costrad[nod])
            costrad[nod]+=costrad[e];
        }
    /*
    fout<<nod<<"\n";
    for(int i=1;i<=n;i++)
        fout<<costrad[i]<<" ";
    fout<<"\n";
    */
}

int main()
{
    int i,x,y,mx=-10000;
    fin>>n;
    for(i=1;i<=n;i++)
        fin>>cost[i];
    for(i=1;i<=n-1;i++)
    {
        fin>>x>>y;
        l[x].push_back(y);
        l[y].push_back(x);
    }
    costrad[1]=cost[1];
    Dfs(1);
    for(i=1;i<=n;i++)
        mx=max(mx,costrad[i]);
    fout<<mx;
    return 0;
}