Cod sursa(job #3247348)

Utilizator Fan_ClubAnime Enjoyer Fan_Club Data 7 octombrie 2024 11:53:37
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.82 kb
#pragma GCC optimize("O3,unroll-loops")
#include <bits/stdc++.h>

using namespace std;
const int NMAX = 16005;
int v[NMAX+5];
vector<int>G[NMAX+5];
int viz[NMAX+5];
long long maxx=LLONG_MIN;
long long F(int nod){
    long long sum=v[nod];viz[nod]=1;
    for(int i=0;i<G[nod].size();i++){
        if(viz[G[nod][i]]==0){
            long long ceva=F(G[nod][i]);
            if(ceva>=0){
                sum+=ceva;
            }
        }
    }
    maxx=max(maxx,sum);
    return sum;
}
int main()
{
    ifstream Klaus("asmax.in");
    ofstream Iohannis("asmax.out");
    int n,a,b;
    Klaus>>n;
    for(int i=1;i<=n;i++){
        Klaus>>v[i];
    }
    for(int i=1;i<n;i++){
        Klaus>>a>>b;
        G[a].push_back(b);
        G[b].push_back(a);
    }
    F(1);Iohannis<<maxx<<'\n';
    return 0;
}