Cod sursa(job #2671168)

Utilizator Ionut2791Voicila Ionut Marius Ionut2791 Data 11 noiembrie 2020 16:42:52
Problema Asmax Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <limits.h>
#include <algorithm>
using namespace std;

const int N = 16005;

int n;
vector<int> graf[N];
int cost[N];
int sp[N];
bool amfost[N];

int dfs(int nod) {
    int sum = 0;

    amfost[nod] = true;
    for (int i = 0; i < (int)graf[nod].size(); ++i) {

        if(amfost[graf[nod][i]] == false)
            sum += dfs(graf[nod][i]) + cost[graf[nod][i]];

    }
    amfost[nod] = false;

    return sum;
}

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

    int n;
    fin >> n;

    for (int i = 1; i <= n; ++i)
        fin >> cost[i];

    for (int i = 1; i < n; ++i) {
        int a, b;
        fin >> a >> b;
        graf[a].push_back(b);
        graf[b].push_back(a);
    }
    dfs(1);

    int maxx = 0;
    for (int i = 1; i<= n; ++i) {
        maxx = max(maxx, dfs(i));
       // cout << dfs(i) << '\n';
    }

    fout << maxx << '\n';

    return 0;
}