Cod sursa(job #2254587)

Utilizator marcudanfDaniel Marcu marcudanf Data 5 octombrie 2018 16:41:44
Problema Diametrul unui arbore Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.71 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

constexpr const int NMAX = 1e5 + 5;

using namespace std;

ifstream fin("darb.in");
ofstream fout("darb.out");

vector < int > g[NMAX];
bool viz[NMAX];
int dist[NMAX];
int n;

pair <int, int> bfs(int nod){
	viz[0] = !viz[0];
	queue < int > q;
	q.push(nod);
	dist[nod] = 1;
	viz[nod] = viz[0];
	while(!q.empty()){
		nod = q.front();
		q.pop();
		for(int i : g[nod])
			if(viz[i] != viz[0]){
				q.push(i);
				viz[i] = 1;
				dist[i] = dist[nod] + 1;
			}
	}
	return {nod, dist[nod]};
}

int main(){
	fin >> n;
	for(int i = 1, x, y; i < n; i++){
		fin >> x >> y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
	fout << bfs(bfs(1).first).second;
	return 0;
}