Cod sursa(job #2940276)

Utilizator MrPuzzleDespa Fabian Stefan MrPuzzle Data 15 noiembrie 2022 10:48:15
Problema Asmax Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.12 kb
#include<fstream>
#include<iostream>
#include<climits>
#include<algorithm>
#include<cstring>
#include<cmath>
#include <vector>

#define MAX 100000

using namespace std;

ifstream f("asmax.in");
ofstream g("asmax.out");

//ifstream f("in.in");
//ofstream g("out.out");

vector <int> l[MAX+5];
int n,m,x,y;
int viz[MAX+5];
int val[MAX+5];
int s[MAX+5];
int maxNod[MAX+5];
int sum=0;
int solMax=-1;

void dfs(int nod){

    s[nod]=val[nod];
    viz[nod] = 1;

    for(int i=0;i<l[nod].size();i++){
        int fiu = l[nod][i];
        if(viz[fiu] == 0){
            dfs(fiu);
            s[nod]+=s[fiu];
            maxNod[nod] = max(maxNod[nod], s[fiu]);
        }
    }

    maxNod[nod] = max(maxNod[nod],(sum-s[nod]));

}

int main(){

    f>>n;
    for(int i=1;i<=n;i++){
        f>>val[i];
        sum+=val[i];
    }
    for(int i=1;i<n;i++){
        f>>x>>y;
        l[x].push_back(y);
        l[y].push_back(x);
    }

    dfs(1);

    for(int i=1;i<=n;i++){
        solMax = max(solMax,maxNod[i]);
    }

    g<<solMax;

    f.close();
    g.close();
    return 0;
}