Cod sursa(job #2693760)

Utilizator Gota_AndreiGota Andrei Gota_Andrei Data 6 ianuarie 2021 22:15:09
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.79 kb
#include<bits/stdc++.h>
using namespace std;

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

int n, x, y, i, nivel_maxim, pozitie_nivel;
bool vizitat[100005];
vector < int > muchii[100005];

void DFS(int nod, int nivel) {
  vizitat[nod] = true;

  if (nivel > nivel_maxim) {
    nivel_maxim = nivel;
    pozitie_nivel = nod;
  }

  for (unsigned int i = 0; i < muchii[nod].size(); i++) {
    int vecin = muchii[nod][i];
    if (!vizitat[vecin])
      DFS(vecin, nivel+1);
  }
}

void reset() {
  for (i = 1; i <= n; i++)
    vizitat[i] = false;
}

int main() {
  fin >> n;
  for (i = 1; i < n; i++) {
    fin >> x >> y;
    muchii[x].push_back(y);
    muchii[y].push_back(x);
  }

  DFS(1, 0);

  reset();

  DFS(pozitie_nivel, 0);

  fout << nivel_maxim + 1;
}