Cod sursa(job #2830327)

Utilizator cosmin1812Nedelcu Adrian Cosmin cosmin1812 Data 9 ianuarie 2022 19:18:11
Problema Asmax Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.09 kb
#include <iostream>
#include<string>
#include<vector>
#include<fstream>
#include<list>
using namespace std;


void dfs_asmax(int nr_noduri, vector<int>v, vector<vector<int>>lista, vector<bool>& vizit, int nod, int& maxi) {

	vizit[nod] = 1;


	for (int i = 1; i < lista[nod].size(); i++) {

		if (vizit[lista[nod][i]] == false) {

			vizit[lista[nod][i]] = true;

			dfs_asmax(nr_noduri, v, lista, vizit, lista[nod][i], maxi);

			if (v[lista[nod][i]] > 0) {

				maxi += v[lista[nod][i]];
			}
		}
	}
}
void asmax() {

	fstream f("asmax.in");
	int n;//nr noduri
	f >> n;

	vector<int>v(n + 1, 0);//valorile nodurilor
	for (int i = 1; i <= n; i++) {

		f >> v[i];
	}

	int x, y;
	vector<vector<int>>lista;//lista de muchii
	lista.resize(n + 1);
	while (f >> x >> y) {

		lista[x].push_back(y);
		lista[y].push_back(x);
	}

	f.close();

	vector<bool> vizit(n + 1, 0);//vector de vizite

	int maxi = 0;//suma noului subarbore

	dfs_asmax(n, v, lista, vizit, 1, maxi);

	ofstream g("asmax.out");
	g << maxi;
	g.close();
}


int main()
{
	asmax();
}