Pagini recente » Cod sursa (job #2027099) | Cod sursa (job #570596) | Cod sursa (job #1637877) | Cod sursa (job #2364973) | Cod sursa (job #2916019)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <climits>
#include <bitset>
using namespace std;
ifstream fin("asmax.in");
ofstream fout("asmax.out");
const int MAX_SIZE = 16005;
bitset<MAX_SIZE> isIterated;
int peakCost[MAX_SIZE], sums[MAX_SIZE];
vector<int> graph[MAX_SIZE];
void findMaxSum(int curPoint) {
sums[curPoint] = peakCost[curPoint];
isIterated[curPoint] = 1;
for (int nextPoint : graph[curPoint]) {
if (isIterated[nextPoint] == 0) {
findMaxSum(nextPoint);
if (isIterated[nextPoint] == 1 && sums[nextPoint] > 0) {
sums[curPoint] += sums[nextPoint];
}
}
}
}
int main() {
int peaks;
fin >> peaks;
for (int i = 1; i <= peaks; ++i) {
fin >> peakCost[i];
}
for (int i = 1; i < peaks; ++i) {
int start, end;
fin >> start >> end;
graph[start].push_back(end);
graph[end].push_back(start);
}
findMaxSum(1);
int maxSum = -INT_MAX;
for (int i = 1; i <= peaks; ++i) {
maxSum = max(maxSum, sums[i]);
}
fout << maxSum;
return 0;
}