Cod sursa(job #961584)

Utilizator addy01adrian dumitrache addy01 Data 12 iunie 2013 17:16:34
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.83 kb
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
vector <int> graf[16010];
bool viz[16010];
int best[16010],v[16010];
void dfs(int x)
{
    viz[x]=1;
    best[x]=v[x];
    for(auto i : graf[x])
        if(!viz[i])
           {
                dfs(i);
                if(best[i]>0)
                best[x]+=best[i];
            }
}

int main()
{
    freopen("asmax.in","r",stdin);
    freopen("asmax.out","w",stdout);
    int n,m,a,b,ans=0;
    cin>>n;
    m=n-1;
    for(int i=1;i<=n;i++)
        cin>>v[i];
    while(m--)
    {
        cin>>a>>b;
        graf[a].push_back(b);
        graf[b].push_back(a);
    }

            dfs(1);
             ans=best[1];
       for(int i=2;i<=n;i++)
        if(best[i]>ans)
            ans=best[i];
        cout<<ans;

    return 0;
}