Cod sursa(job #2531725)

Utilizator ioanavasi16Vasile Ioana ioanavasi16 Data 26 ianuarie 2020 17:48:10
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <fstream>
#include <vector>
#include <algorithm>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
vector <int> a[16005];
int M,N,viz[16005],d[16005],v[16005],Maxim;
void DFS(int nod)
{
    viz[nod]=1;
    d[nod]=v[nod];
    for(int i=0; i<a[nod].size(); i++)
    {
        int vec=a[nod][i];
        if(viz[vec]==0)
        {
            DFS(vec);
            d[nod]=max(d[nod],d[nod]+d[vec]);
        }
        Maxim=max(Maxim,d[nod]);
    }
}
void creare()
{
    for(int i=1; i<N; i++)
    {
        int x,y;
        fin>>x>>y;
        a[x].push_back(y);
        a[y].push_back(x);
    }
}
int main()
{
    fin>>N;
    for(int i=1;i<=N;i++)
    {
        fin>>v[i];
    }
    creare();
    DFS(1);
    fout<<Maxim;
    return 0;
}