Pagini recente » Cod sursa (job #554195) | Monitorul de evaluare | Cod sursa (job #1364517) | Cod sursa (job #754465) | Cod sursa (job #3357016)
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
vector<vector<int>>tree;
vector<int>subTree;
vector<int>values;
int maxi = -1000;
void DFS(int crt, int parent) {
subTree[crt] += values[crt];
for (auto i : tree[crt]) {
if (i != parent) {
DFS(i, crt);
subTree[crt] += max(0,subTree[i]);
}
}
maxi = max(maxi, subTree[crt]);
}
int main()
{
int n;
fin >> n;
values.resize(n);
tree.resize(n);
subTree.resize(n);
int root = 0;
for (int i = 0; i < n; ++i) {
fin >> values[i];
}
int u, v;
for (int i = 0; i < n - 1; ++i) {
fin >> u >> v;
--u;
--v;
tree[u].push_back(v);
tree[v].push_back(u);
}
DFS(0, -1);
fout << maxi;
return 0;
}