Cod sursa(job #2706307)

Utilizator AokijiAlex M Aokiji Data 14 februarie 2021 16:01:52
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.95 kb
#include <fstream>
#include <algorithm>

using namespace std;

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

int const NM = 16001;
int v[NM],vf[1+2*NM],nxt[1+2*NM],val[NM],dp[NM];
int sz;
bool viz[NM];

void add (int x , int y)
{
    ++ sz;
    vf[sz] = y;
    nxt[sz] = v[x];
    v[x] = sz;
}

void dfs (int node)
{
    viz[node] = true;
    for(int i=v[node];i;i=nxt[i])
    {
        int y = vf[i];
        if (!viz [y])
        {
            dfs(y);
            dp[node] = max (dp[node]+dp[y] , dp[node]);
        }
    }
}
int main()
{
    int n;
    fin>>n;
    for(int i=1 ; i<=n ; i++)
        fin>>val[i];
    for(int i=1 ; i<n ; i++)
    {
        int a,b;
        fin>>a>>b;
        add(a,b);
        add(b,a);
    }
    for(int i=1 ; i<=n ; i++)
        dp[i] = val[i];
    for (int i=1 ; i<=n ; i++)
        if (! viz [i])
            dfs(i);
    fout<<*max_element (dp+1 , dp+1+n);
    return 0;
}