Cod sursa(job #2776735)

Utilizator RamanujanNeacsu Mihnea Ramanujan Data 20 septembrie 2021 21:25:47
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <bits/stdc++.h>
#define MAXN 100001
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");
vector<int> g[MAXN];
int dist[MAXN];
void bfs(int i){
   queue<int> q;
   q.push(i);
   int copy=i;
   while(!q.empty()){
       i=q.front();
       for(int j=0; j<g[i].size(); j++){
          if(dist[g[i][j]]==0){
            dist[g[i][j]]=dist[i]+1;
            q.push(g[i][j]);
          }
       }
       q.pop();
   }
   dist[copy]=0;
}
int main()
{
    int n, hi=0, newTarget; fin>>n;
    for(int i=1; i<n; i++){
       int x, y; fin>>x>>y;
       g[x].push_back(y);
       g[y].push_back(x);
    }
    bfs(1);
    for(int i=1; i<=n; i++){
        if(dist[i]>hi){
           hi=dist[i];
           newTarget=i;
        }
    }
    memset(dist, 0, sizeof(dist));
    bfs(newTarget);
    hi=0;
    for(int i=1; i<=n; i++){
        if(dist[i]>hi){
           hi=dist[i];
        }
    }
    fout<<hi+1;
    return 0;
}