Pagini recente » Cod sursa (job #3202874) | Cod sursa (job #1033619) | Cod sursa (job #2238347) | Cod sursa (job #1814946) | Cod sursa (job #1242277)
#include <fstream>
#include <vector>
#include <algorithm>
#include <bitset>
using namespace std;
ifstream is("asmax.in");
ofstream os("asmax.out");
void Read();
void Dfs(int k);
int n, m;
vector<vector<int>> g;
bitset<16001> v;
vector<int> smax;
int main()
{
Read();
Dfs(1);
int maxim = -0x3f3f3f3f;
for(int i = 1; i <= n; ++i)
{
maxim = max(maxim, smax[i]);
}
os << maxim;
is.close();
os.close();
return 0;
}
void Read()
{
is >> n;
g = vector <vector<int>>(n+1);
// v = vector<bool>(n + 1);
smax = vector<int>(n + 1);
int x, y;
for(int i = 1; i <= n; ++i)
{
is >> x;
smax[i] = x;
}
for(int i = 1; i <= n-1; ++i)
{
is >> x >> y;
g[x].push_back(y);
g[y].push_back(x);
}
}
void Dfs(int k)
{
v[k] = true;
for (const auto& x : g[k])
{
if( v[x]) continue;
Dfs(x);
if(smax[x] > 0 )
smax[k] += smax[x];
}
}