Cod sursa(job #1962082)

Utilizator ioana_99Ioana Marin ioana_99 Data 11 aprilie 2017 16:01:25
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <stdio.h>
#include <vector>
using namespace std;
int cost[16010];
vector <int> a[16010];
bool marcat[16010];
void dfs(int nod)
{
    int s=0;
    marcat[nod]=true;
    for(int i=0; i<a[nod].size(); i++)
        if(!marcat[a[nod][i]])
            {
                dfs(a[nod][i]);
                if(cost[a[nod][i]]>0)s=s+cost[a[nod][i]];
            }
    cost[nod]+=s;
}
int maxi=-2000000000, n, x, y;
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", &x, &y);
        a[x].push_back(y);
        a[y].push_back(x);
    }
    dfs(1);
    for(int i=1; i<=n; i++)
    {
        if(maxi<=cost[i]) maxi=cost[i];
    }
    printf("%d", maxi);
    return 0;
}