Cod sursa(job #3236359)

Utilizator Dia3141Costea Diana Stefania Dia3141 Data 28 iunie 2024 11:17:14
Problema Asmax Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.69 kb
#include <fstream>
#include <vector>
#include <bitset>
#define dim 16002
using namespace std;
ifstream cin("asmax.in");
ofstream cout("asmax.out");
int n,dp[dim],x,y,root,sol;
bitset<dim>grad;
vector<int>v[dim];
void dfs(int x){
    for(auto i: v[x]){
        dfs(i);
        dp[x]=max(dp[x],dp[x]+dp[i]);
        sol=max(sol,dp[x]);
    }
}
int main()
{
    cin>>n;
    sol=-10000;
    for(int i=1;i<=n;i++)
        cin>>dp[i];
    for(int i=1;i<n;i++){
        cin>>x>>y;
        v[x].push_back(y);
        grad[y]=1;
    }
    for(int i=1;i<=n;i++)
        if(!grad[i]){
            root=i;
            break;
        }
    dfs(root);
    cout<<sol;
    return 0;
}