Cod sursa(job #2953665)

Utilizator Zed1YasuoAlex Birsan Zed1Yasuo Data 11 decembrie 2022 22:06:40
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
int n,m,s[16001],mx=-1e9,b[16001],viz[16003];
vector<int>a[16001];
void dfs(int nod)
{
    viz[nod]=1;
    s[nod]=b[nod];
    for(int i=0;i<a[nod].size();i++)
    {
        int vec=a[nod][i];
        if(viz[vec]==0)
        {
            dfs(vec);
            if(s[vec]>0)
                s[nod]+=s[vec];
        }
    }
    mx=max(mx,s[nod]);
}
int main()
{
    f>>n;
    for(int i=1;i<=n;i++)
        f>>b[i];
    for(int i=1;i<n;i++)
    {
        int x,y;
        f>>x>>y;
        a[x].push_back(y);
        a[y].push_back(x);
    }
    dfs(1);
    g<<mx;
    return 0;
}