Pagini recente » Cod sursa (job #96347) | Cod sursa (job #3246226) | Cod sursa (job #1535456) | Cod sursa (job #3017128) | Cod sursa (job #2913766)
#include <bits/stdc++.h>
#define ll long long
using namespace std;
ifstream reader("asmax.in");
ofstream writer("asmax.out");
const int inf = 1e9 + 12415;
void dfs (int k, vector <int>& rez, int poz, vector <int>& copac, vector <int>& a)
{
if(copac[k] == 0)
return;
//rez[k] += a[k];
rez[copac[k]] += rez[k];
dfs(copac[k], rez, poz, copac, a);
}
int main()
{
// #ifndef ONLINE_JUDGE
// freopen("input", "r", stdin);
// freopen("outputs", "w", stdout);
// #endif
int n;
reader >> n;
vector <int> a(n + 1);
vector <bool> rad(n + 1, false);
vector <int> copac(n + 1 , inf);
for(int i = 1; i <= n; ++i)
reader >> a[i];
for(int i = 1; i < n; ++ i)
{
int x, y;
reader >> x >> y;
rad[x] = true;
copac[y] = x;
}
int poz = -1;
for(int i = 0; i <= n; ++i)
if(copac[i] == inf)
{
copac[i] = 0;
poz = i;
}
vector <int> rez(n + 1, 0);
for(int i = 0; i <= n; ++i)
rez[i] = a[i];
for(int i = 1; i <= n; ++i)
{
if(!rad[i])
{
int ant = -1;
dfs(i, rez, poz, copac, a);
}
}
int maxi = -1e9;
int cnt = 0;
for(int i = 0; i <= n; ++i)
{
if(rez[i] > maxi)
maxi = rez[i];
if(copac[i] == poz and rez[i] + a[poz] > maxi)
{
maxi = rez[i] + a[poz];
cnt += rez[i];
}
}
if(cnt + a[poz] > maxi)
maxi = cnt + a[poz];
writer << maxi;
return 0;
}