Cod sursa(job #1708814)

Utilizator andru47Stefanescu Andru andru47 Data 27 mai 2016 23:36:48
Problema Asmax Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.9 kb
#include <bits/stdc++.h>
#define pb push_back
using namespace std;
const int NMAX = 16000 + 5;
int n,i,cost[NMAX],a,b,ret;
vector <int> g[NMAX];
bool sel[NMAX];
inline void dfs(int x, int &s)
{
    sel[x] = true;
    for(int i = 0; i<g[x].size(); ++i)
        if (!sel[g[x][i]])
        {
            s+=cost[g[x][i]];
            ret = max(ret,s);
            dfs(g[x][i],s);
        }
}
int main()
{
    freopen("asmax.in","r",stdin);
    freopen("asmax.out","w",stdout);
    scanf("%d", &n);
    for (int i = 1; i<=n ; ++i)
        scanf("%d", &cost[i]);
    for (int i = 1; i<n; ++i)
    {
        scanf("%d %d\n", &a, &b);
        g[a].pb(b);
        g[b].pb(a);
    }
    for (int i = 1; i<=n; ++i)
    {
        if (!sel[i])
        {
            int s = 0;
            dfs(i,s);
            ret = max(ret,s);
        }
    }
    printf("%d\n", ret);
    return 0;
}