Cod sursa(job #2781809)

Utilizator TudosieRazvanTudosie Marius-Razvan TudosieRazvan Data 10 octombrie 2021 15:27:13
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <fstream>
#include <vector>
using namespace std;

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

int n;
int costuri[16001],vizitat[16001];
vector<int> noduri[16001];
int maxim;

void dfs(int k)
{
    vizitat[k]=1;
    for(int i=0; i<noduri[k].size(); i++)
    {
        int nod=noduri[k][i];
        if(vizitat[nod]==0)
        {
            //nu am vizitat nodul
            dfs(nod);
            if(costuri[k]<costuri[nod]+costuri[k])
            {
                costuri[k]=costuri[nod]+costuri[k];
            }

        }
    }
    maxim=max(maxim,costuri[k]);
}

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

    for(int i=1; i<=n; i++)
    {
        int x,y;
        fin>>x>>y;
        noduri[x].push_back(y);
        noduri[y].push_back(x);
    }

    dfs(1);
    fout<<maxim;
    return 0;
}