Cod sursa(job #2503163)

Utilizator YetoAdrian Tonica Yeto Data 2 decembrie 2019 16:53:04
Problema Asmax Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.84 kb
#include <fstream>
#include <vector>
#include <cstring>
#define DIM 16010
using namespace std;
ifstream fin ("asmax.in");
ofstream fout("asmax.out");
int n, x, y, maxim, j, i;
int viz[DIM], S[DIM], val[DIM];
vector <int> g[DIM];

void dfs(int nod) {
    viz[nod] = 1;
    S[nod] = val[nod];
    for (int i=0;i<g[nod].size(); i++) {
        int vecin = g[nod][i];
        if (viz[vecin] == 0) {
            dfs(vecin);
            S[nod] += S[vecin];
        }
    }
}

int main () {
    fin>>n;
    for (i=1;i<=n;i++) {
        fin>>val[i];
    }
    for (i=1;i<n;i++) {
        fin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    for (i=1;i<=n;i++) {
        dfs(i);
        for (j=1;j<=n;j++) {
            if (S[j]>maxim)
                maxim=S[j];
        }
        memset(viz, 0, sizeof(viz));
    }
    fout<<maxim;
    return 0;
}