Cod sursa(job #1880156)

Utilizator RaZxKiDDavid Razvan RaZxKiD Data 15 februarie 2017 16:22:58
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#include <cstring>

using namespace std;

int D[100001],n;
vector<int> L[100001];

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

int bfs(int x){
    int imax,dmax=0;
    memset(D,0,sizeof(D));
    D[x]=1;
    queue<int> Q;
    Q.push(x);
    int a;
    while(!Q.empty()){
        a=Q.front();
        Q.pop();
        for(auto vecin : L[a]){
            if(D[vecin]==0){
                D[vecin]=D[a]+1;
                Q.push(vecin);
                if(dmax<D[vecin]){
                    dmax=D[vecin];
                    imax=vecin;
                }
            }
        }
    }
    return imax;
}
void read(){
    in>>n;
    int x,y;
    for(int i=1;i<n;i++){
        in>>x>>y;
        L[x].push_back(y);
        L[y].push_back(x);
    }
}
int main(){
    read();
    out<<D[bfs(bfs(1))];
    return 0;
}