Cod sursa(job #2780221)

Utilizator mihnea.cazan15mihnea cazan mihnea.cazan15 Data 6 octombrie 2021 15:02:18
Problema Asmax Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.64 kb
#include <bits/stdc++.h>
#include <fstream>
using namespace std;
ifstream f("asmax.in");
ofstream g("asmax.out");
/*const long long mod = 998244353;
long long fct[200005],invfct[200005],inv[200005];
long long put2[200005];
long long lgput (long long a, long long exp)
{
    long long rz=1;
    while(exp)
        if(exp&1)
            exp^=1,rz=rz*1LL*a%mod;
        else
            exp>>=1,a=a*1LL*a%mod;
    return rz;
}
long long cmb (long long n, long long k)
{
    if(n<k || k<0 || n<0)
        return 0;
    return fct[n]*1LL*invfct[k]%mod*1LL*invfct[n-k]%mod;
}
void init ()
{
    inv[1]=fct[0]=fct[1]=invfct[0]=invfct[1]=put2[0]=1,put2[1]=2;
    for(long long i=2;i<=200000;++i)
        put2[i]=put2[i-1]*2LL%mod,inv[i]=(mod-mod/i)*1LL*inv[mod%i]%mod,fct[i]=i*1LL*fct[i-1]%mod,invfct[i]=inv[i]*1LL*invfct[i-1]%mod;
}
long long v[200005];*/
const int inf=(-1)*(1<<30);
int rz[16001];
vector<int> v[16001];
void dfs(int fiu,int tata)
{
    for(int i=0;i<v[fiu].size();i++)
    {
        int vecin=v[fiu][i];
        if(vecin!=tata)
        {
            dfs(vecin,fiu);
            rz[fiu]=max(rz[fiu],rz[fiu]+rz[vecin]);
        }
    }
}
void solve()
{
    int n;
    f>>n;
    for(int i=1;i<=n;i++)
        f>>rz[i];
    for(int i=1;i<n;i++)
    {
        int a,b;
        f>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    //cout<<"hatz";
    dfs(1,0);
    int ans=inf;
    for(int i=1;i<=n;i++)
        ans=max(ans,rz[i]);
    g<<ans;
}
int main()
{
    /*ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int t=1;
    while(t--)
    */solve();
    return 0;
}