Cod sursa(job #2500525)

Utilizator mihaimodiMihai Modi mihaimodi Data 28 noiembrie 2019 09:45:59
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int n;
vector <int> g[16001];
int v[16001],d[16001];
bool viz[16001];
void dfs(int nod)
{
    viz[nod]=1;
    for(int i=0;i<g[nod].size();i++)
    {
        int vecin=g[nod][i];
        if(viz[vecin]==0)
        {
            dfs(vecin);
            if(d[nod]<d[vecin]+d[nod])
                d[nod]=d[vecin]+d[nod];
        }
    }
}
int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
    {
        fin>>v[i];
        d[i]=v[i];
    }
    for(int i=1;i<n;i++)
    {
        int x, y;
        fin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1);
    int Max=-1001;
    for(int i=1;i<=n;i++)
        Max=max(Max,d[i]);
    fout<<Max;
    return 0;
}