Cod sursa(job #3223369)

Utilizator T1raduTaerel Radu Nicolae T1radu Data 13 aprilie 2024 11:05:51
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
int n,sum[16001],val[16001],viz[16001];
vector<int> G[16001];
void dfs(int x)
{
    viz[x]=1;
    for(vector<int>::iterator it=G[x].begin();it!=G[x].end();++it)
    {
        if(!viz[*it])
        {
            dfs(*it);
            sum[x]=max(sum[x],sum[x]+sum[*it]);
        }
    }
}
int main()
{
    fin >> n;
    for(int i=1;i<=n;i++)
    {
        fin >> val[i];
        sum[i]=val[i];
    }
    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        fin >> x >> y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    dfs(1);
    int maxim=0;
    for(int i=1;i<=n;i++)
    {
        maxim=max(maxim,sum[i]);
    }
    fout << maxim;
    return 0;
}