Cod sursa(job #2952619)

Utilizator NiffSniffCojocaru Calin Marcu NiffSniff Data 9 decembrie 2022 16:37:43
Problema Asmax Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <fstream>
#include <vector>
#include <bitset>
using namespace std;
const int N = 1000000;
string file = "asmax";
ifstream cin(file +".in");
ofstream cout(file +".out");
vector <int> a[N + 1],v,scor;
bitset <N + 1> viz;
int n, cnt, smax(-1000);


void citire()
{
	int x, y;
	cin >> n;
	scor.resize(n+1);
	v.push_back(0); // v[0] = 0;
	for (int i = 1; i <= n; i++)
	{
		cin >> x;
		v.push_back(x);
	}
	for (int i = 1; i < n; i++)
	{
		cin >> x >> y;
		a[x].push_back(y);
		a[y].push_back(x);
	}
}

void DFS(int x)
{
	viz[x] = 1;
	scor[x] = v[x];
	for (int y : a[x])
	{
		if (!viz[y])
		{
			DFS(y);
			if (scor[y] > 0)
			{
				scor[x] += scor[y];
			}
		}
	}
}
int main()
{
	citire();
	DFS(1);
	for (int i : scor)
	{
		if (smax < i)
			smax = i;
	}
	cout << smax;
}