Cod sursa(job #1889676)

Utilizator UrsuDanUrsu Dan UrsuDan Data 22 februarie 2017 20:38:09
Problema Asmax Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <cstdio>
#include <vector>

using namespace std;

vector< vector<int> >g(16010);

int v[16010];
bool viz[16010];

void dfs(int node)
{
    int i;
    viz[node]=true;
    for(i=0; i<g[node].size(); i++)
    {
        if(viz[g[node][i]]==false)
        {
            dfs(g[node][i]);
            if(v[g[node][i]]>0)
                v[node]+=v[g[node][i]];
        }
    }
}

int main()
{
    freopen("asmax.in","r",stdin);
    freopen("asmax.out","w",stdout);
    int i,n,node,x,y;
    scanf("%d",&n);
    node=1;
    for(i=1; i<=n; i++)
    {
        scanf("%d",&v[i]);
        if(v[i]>v[node])
            node=i;
    }
    for(i=1; i<=n-1; i++)
    {
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(node);
    //printf("///////////");
    printf("%d",v[node]);
    return 0;
}