Cod sursa(job #2434103)

Utilizator ShayTeodor Matei Shay Data 30 iunie 2019 16:23:29
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.13 kb
#include <fstream>
#include <vector>
#include <queue>
#include <time.h>

void bfs(int src, int n, int &last_node, int &diametru, std::vector<int> distance, std::vector<int> *adj) {
	std::vector<bool> visited(n, false);
	std::queue<int> Q;
	Q.push(src);
	visited[src] = true;

	while (!Q.empty()) {
		int index = Q.front();
		Q.pop();

		for (auto it = adj[index].begin() ; it != adj[index].end() ; ++it) {
			if (visited[*it]) {
				continue;
			}

			distance[*it] = distance[index] + 1;
			Q.push(*it);
			visited[*it] = true;
			last_node = *it;
			diametru = distance[*it];
		}
	}
}

int main() {
	std::ifstream cin("darb.in");
	std::ofstream cout("darb.out");
	std::ios::sync_with_stdio(false);
	srand((unsigned)time(NULL));
	int n, last_node, src, dest, diametru;
	cin >> n;
	std::vector<int> adj[n];
	std::vector<int> distance(n, 0);

	for (int i = 0 ; i < n - 1 ; ++i) {
		cin >> src >> dest;
		adj[src - 1].push_back(dest - 1);
		adj[dest - 1].push_back(src - 1);
	}
	int node = rand() % (n - 1);
	bfs(node, n, last_node, diametru, distance, adj);
	bfs(last_node, n, last_node, diametru, distance, adj);

	cout << diametru  + 1 << '\n';
	return 0;
}