Cod sursa(job #1913085)

Utilizator andreigasparoviciAndrei Gasparovici andreigasparovici Data 8 martie 2017 11:44:35
Problema Asmax Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <iostream>
#include <vector>
#include <cstdio>
#include <bitset>
#include <algorithm>
#define NMAX 16005
using namespace std;

int n, v[NMAX];
vector<int> arb[NMAX];
bitset<NMAX> viz;

int result=-0x3f3f3f3f;

int max3(int a,int b,int c)
{
    return max(a,max(b,c));
}

void dfs(int x, int parent)
{
    result = max(result,v[x]+result);
    for(vector<int>::iterator it = arb[x].begin();it!=arb[x].end();it++)
    {
        if(result+v[*it] > result)
            dfs(*it,x);
    }
}

int main()
{
    freopen("asmax.in","r",stdin);
    freopen("asmax.out","w",stdout);
    scanf("%d ",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d ",&v[i]);
    }
    for(int i=1;i<n;i++)
    {
        int a, b;
        scanf("%d %d ",&a,&b);
        arb[a].push_back(b);
        arb[b].push_back(a);
    }
    result=v[1];
    dfs(1,0);
    printf("%d",result);
    return 0;
}