Cod sursa(job #1922571)

Utilizator greenadexIulia Harasim greenadex Data 10 martie 2017 17:54:56
Problema Asmax Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.86 kb
#include <bits/stdc++.h>

#define pb push_back
#define f first
#define s second
#define pii pair<int, int>
#define mp make_pair
 
using namespace std;
 
const string name = "asmax",
             in_file = name + ".in",
             out_file = name + ".out";
 
ifstream fin(in_file);
ofstream fout(out_file);
 
const int MAX = 1.6e4 + 5;

int n;
int sum[MAX];
vector<int> tree[MAX];

void dfs(int node, int father) {
	for (int adj : tree[node]) {
		if (adj != father) {
			dfs(adj, node);
			if (sum[adj] > 0) {
				sum[node] += sum[adj];
			}
		}
	}
}

int main() {
	fin >> n;
	for (int i = 1; i <= n; i++) {
		fin >> sum[i];
	}

	for (int a, b, i = 1; i < n; i++) {
		fin >> a >> b;
		tree[a].push_back(b);
		tree[b].push_back(a);
	}

	dfs(1, 1);

	int max_sum = -(1 << 30);
	for (int i = 1; i <= n; i++) {
		max_sum = max(max_sum, sum[i]);
	}
	
	fout << max_sum;
	return 0;
}