Cod sursa(job #3301045)

Utilizator adimiclaus15Miclaus Adrian Stefan adimiclaus15 Data 21 iunie 2025 10:57:01
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream cin("asmax.in");
ofstream cout("asmax.out");
int n, a, b, maxx;
int v[16002], dp[16002], viz[16002];
vector<int> L[16002];
void dfs(int nod){
    dp[nod]=v[nod];
    viz[nod]=1;
    vector<int> fii;
    for(int i=0;i<L[nod].size();i++){
        int vecin=L[nod][i];
        if(!viz[vecin]){
            dfs(vecin);
            fii.push_back(vecin);
            // if(dp[vecin]>0)
            //     dp[nod]+=dp[vecin];
        }

    }
    for(auto fiu : fii) {
        if(dp[fiu] > 0) {
            dp[nod] += dp[fiu];
        }
    }
}
int main() {
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>v[i];
    }
    for(int i=1;i<n;i++){
        cin>>a>>b;
        L[a].push_back(b);
        L[b].push_back(a);
    }
    dfs(1);
    maxx=-1e9;
    for(int i=1;i<=n;i++){
        maxx=max(maxx, dp[i]);
       //cout<<dp[i]<<" ";
    }
    cout<<maxx;
}