Pagini recente » Cod sursa (job #1702852) | Cod sursa (job #1591271) | Cod sursa (job #13760) | Cod sursa (job #624239) | Cod sursa (job #2539748)
#include <bits/stdc++.h>
using namespace std;
const int NMAX = 16006;
int N, cost[NMAX];
vector <int> graph[NMAX];
int dp[NMAX], answer;
void dfs(int node, int father)
{
dp[node] = cost[node];
for (auto& x : graph[node])
if (x != father)
{
dfs(x, node);
if (dp[x] > 0)
dp[node] += dp[x];
}
answer = max(answer, dp[node]);
}
int main()
{
ifstream fin("asmax.in");
ofstream fout("asmax.out");
fin >> N;
answer = -2e9;
for (int i = 1;i <= N;++i)
{
fin >> cost[i];
answer = max(answer, cost[i]);
}
for (int i = 1;i < N;++i)
{
int u, v;
fin >> u >> v;
graph[u].push_back(v);
graph[v].push_back(u);
}
dfs(1, 0);
fout << answer << "\n";
fin.close();
fout.close();
return 0;
}