Cod sursa(job #1768044)

Utilizator tqmiSzasz Tamas tqmi Data 30 septembrie 2016 08:33:35
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;

ifstream fin("asmax.in");
ofstream fout("asmax.out");
int val[16005],m=-(1<<30),N,dp[16005],use[16005];
vector <int> P[16005];
void read()
{
    fin>>N;
    for(int i=1;i<=N;i++)
        fin>>val[i];
    for(int i=1;i<N;i++)
    {
        int a,b;
        fin>>a>>b;
        P[a].push_back(b);
        P[b].push_back(a);
    }
}
void dfs(int nod)
{
    use[nod]=1;
    dp[nod]=val[nod];
    for(int i=0;i<P[nod].size();i++)
    {
        int vecin=P[nod][i];
        if(!use[vecin])
        {
            dfs(vecin);
            dp[nod]=max(dp[nod],dp[nod]+dp[vecin]);
        }
    }
    m=max(m,dp[nod]);
}
void print()
{
    fout<<m;
}
int main()
{
    read();
    dfs(1);
    print();
    return 0;
}