Cod sursa(job #2662354)

Utilizator Leonard123Mirt Leonard Leonard123 Data 23 octombrie 2020 22:10:56
Problema Asmax Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.66 kb
#include<bits/stdc++.h>
#define pb push_back
using namespace std;
 
ifstream fin("asmax.in");
ofstream fout("asmax.out");

vector <int> fiu[16005];

int val[16005], tata[16005] ,eTata[1005], n, x, y, best = -1000;

void bestSum(int copil) {
	for (auto next  : fiu[copil]) 
		bestSum(next);
	best = max(val[copil], best);  
	val[tata[copil]] = max(val[tata[copil]], val[tata[copil]] + val[copil]);
}
 
int main() {
	fin >> n;	
	for (int i = 1; i <= n; i++)
		fin >> val[i];
	for (int i = 1; i < n; i++) {
		fin >> x >> y;
		tata[y] = x;
		fiu[x].pb(y);
		eTata[y] = 1;
	}	
	for (int i = 1; i <= n; i++) 
		if (eTata[i] == 0) {
			bestSum(i);	
			break;
		}
	fout << best;
}