Cod sursa(job #2803672)

Utilizator andreimocianAndrei Mocian andreimocian Data 20 noiembrie 2021 12:27:12
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.03 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

int n, sol;
int dp[16005], viz[16005];
vector < int > L[16005];

void citire()
{
    int x, y;
    fin >> n;
    for(int i = 1; i <= n; i++)
        fin >> dp[i];
    for(int i = 1; i <= n - 1; i++)
    {
        fin >> x >> y;
        L[x].push_back(y);
        L[y].push_back(x);
    }
}

void DFS(int nod, int tata)
{
    viz[nod] = 1;
    for(int i = 0; i < L[nod].size(); i++)
    {
        int nnod = L[nod][i];
        if(nnod != tata)
        {
            if(viz[nnod] == 0)
            {
                DFS(nnod, nod);
            }
            if(dp[nnod] > 0)
            {
                dp[nod] += dp[nnod];
            }
        }
    }
    sol = max(sol, dp[nod]);
}

int main()
{
    citire();
    for(int i = 1; i <= n; i++)
    {
        if(viz[i] == 0)
        {
            DFS(i,0);
        }
    }
    fout << sol;
    return 0;
}