Cod sursa(job #2109250)

Utilizator MithrilBratu Andrei Mithril Data 19 ianuarie 2018 13:23:51
Problema Asmax Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.59 kb
#include <bits/stdc++.h>
#define NMAX 16001
using namespace std;

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

int n;
int val[NMAX];
bitset<NMAX> vis;
vector<int> graf[NMAX];

int dfs(int i)
{
    vis[i]=1;
    int sum = val[i];
    for(auto q:graf[i])
        if(!vis[q])
            sum=max(sum, sum+dfs(q));
    return sum;
}

int main()
{
    fin>>n;
    for(int i=1;i<=n;i++) fin>>val[i];
    for(int i=1;i<n;i++)
    {
        int x,y;
        fin>>x>>y;
        graf[x].push_back(y);
        graf[y].push_back(x);
    }
    fout<<dfs(1);
    return 0;
}