Cod sursa(job #3362236)

Utilizator Darius9705Darius boros Darius9705 Data 4 august 2026 15:53:28
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
vector<vector<int>> v(16001);
int val[16001],dp[16001],maxi=-2000000000;

void dfs(int nod,int p)
{
    dp[nod]=val[nod];
    for(int i=0;i<v[nod].size();i++)
    {
        int vec=v[nod][i];
        if(vec!=p)
        {
            dfs(vec,nod);
            if(dp[vec]>0)
            {
                dp[nod]+=dp[vec];
            }
        }
    }
    if(dp[nod]>maxi)
    {
        maxi=dp[nod];
    }
}

int main()
{
    int n,i,x,y;
    fin>>n;
    for(i=1;i<=n;i++)
    {
        fin>>val[i];
    }
    for(i=1;i<n;i++)
    {
        fin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1,0);
    fout<<maxi;
    return 0;
}