Cod sursa(job #1801468)

Utilizator amaliarebAmalia Rebegea amaliareb Data 9 noiembrie 2016 00:57:13
Problema Asmax Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.89 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
int n,i,j,v[1601],t[1601],x,y,r,best[1601],maxi;
vector<int> fiu[1601];

void dfs(int nod)
{
    int i;
    for(i=0;i<fiu[nod].size();i++)
    {
        dfs(fiu[nod][i]);
        best[nod]=max(best[nod],best[nod]+best[fiu[nod][i]]);
    }
}

int main()
{
    f>>n;
    for(i=1;i<=n;i++)
    {
        f>>v[i];
        best[i]=v[i];
    }
    for(i=1;i<n;i++)
    {
        f>>x>>y;
        if(!t[x])
        {
            t[x]=y;
            fiu[y].push_back(x);
        }
        else if(!t[y])
        {
            t[y]=x;
            fiu[x].push_back(y);
        }
    }
    for(i=1;i<=n;i++) if(t[i]==0) r=i;
    dfs(r);
    maxi=best[1];
    for(i=1;i<=n;i++) if(best[i]>maxi) maxi=best[i];
    g<<maxi<<'\n';
    return 0;
}