Cod sursa(job #2471943)

Utilizator rebecca0312Andrei Rebecca rebecca0312 Data 11 octombrie 2019 19:41:58
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.84 kb
#include<cstdio>
#include<vector>
using namespace std;

const int NMAX=16005;

bool viz[NMAX];
int dp[NMAX],maxx;
vector<int> G[NMAX];

void dfs(int nod, int tata){
    for(int i=0;i<(int)G[nod].size();i++){
        int fiu=G[nod][i];
        if(fiu!=tata){
            dfs(fiu, nod);
            if(dp[fiu]>0)
                dp[nod]+=dp[fiu];
        }
    }
    maxx=max(maxx, dp[nod]);
    viz[nod]=1;
}

int main(){
    freopen("asmax.in","r",stdin);
    freopen("asmax.out","w",stdout);
    int n;
    scanf("%d", &n);
    for(int i=1;i<=n;i++)
        scanf("%d", &dp[i]);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d", &a, &b);
        G[a].push_back(b);
        G[b].push_back(a);
    }
    for(int i=1;i<=n;i++)
        if(!viz[i])
            dfs(i, 0);
    printf("%d", maxx);
    return 0;
}