Cod sursa(job #3205550)

Utilizator Theo20067Cismaru Theodor-Alexe Theo20067 Data 20 februarie 2024 00:53:31
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.7 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin ("asmax.in");
ofstream fout("asmax.out");
const int INF=1e9;
int n,i,x,y,V[16003],k,maxim,D[16003];
vector <int> L[16003];
void dfs(int nod,int t)
{
    for(int j=0;j<L[nod].size();j++)
        if(L[nod][j]!=t)
            dfs(L[nod][j],nod);
    if(D[t]+D[nod]>D[t]&&nod!=1)
        D[t]+=D[nod];
    maxim=max(max(D[nod],D[t]),maxim);
}
int main()
{
    fin>>n;
    maxim=-INF;
    for(i=1;i<=n;i++)
    {
        fin>>V[i];
        D[i]=V[i];
    }
    for(i=1;i<n;i++)
    {
        fin>>x>>y;
        L[x].push_back(y);
        L[y].push_back(x);
    }
    dfs(1,0);
    fout<<maxim;
    return 0;
}