Cod sursa(job #2294598)

Utilizator vladth11Vlad Haivas vladth11 Data 2 decembrie 2018 16:33:40
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("asmax.in");
ofstream cout("asmax.out");
const int N = 100005;
vector <int>v[N];
bool viz[N];
long long sum[N];
long long maxim = -2000000000;
int n,m;
void dfs(int nod){
    viz[nod] = 1;
    int i;
    for(i = 0;i < v[nod].size();i++){
        if(!viz[v[nod][i]]){
            dfs(v[nod][i]);
            if(sum[v[nod][i]] > 0)
               sum[nod] += sum[v[nod][i]];
        }
    }
    if(sum[nod] > maxim)
        maxim = sum[nod];
}
int main()
{
    int x,y,cnt = 0,i;
    cin >> n;
    for(i = 1;i <= n;i++)
        cin >> sum[i];
    for(i = 1;i <= n-1;i++){
        cin >> x >> y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    for(i = 1;i <= n;i++){
        if(!viz[i]){
            cnt++;
            dfs(i);
        }
    }
    cout << maxim;
    return 0;
}