Cod sursa(job #2371320)

Utilizator mariastStoichitescu Maria mariast Data 6 martie 2019 17:10:05
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.88 kb
#include<fstream>
#include<vector>
#include<queue>
#include<cstring>
#define MAX_N 1000100
using namespace std;
ifstream f ("darb.in");
ofstream g ("darb.out");
vector<int> G[100100];
int viz[100100],n,x,y,ultim,diametru;
queue<int>q;
void bfs(int x){
//    memset(viz,0,MAX_N);
    q.push(x);
    viz[x]=1;
    while(!q.empty()){
        x=q.front();
        for(int i=0;i<G[x].size();++i){
            if(!viz[G[x][i]]){
                viz[G[x][i]]=viz[x]+1;
                q.push(G[x][i]);
                ultim=G[x][i];
                diametru=viz[G[x][i]];
            }
        }
        q.pop();
    }
}
int main()
{
    f>>n;
    for(int i=1;i<n;++i){
        f>>x>>y;
        G[x].push_back(y);
        G[y].push_back(x);
    }
    bfs(1);
    for(int i=1;i<=n;++i){
        viz[i]=0;
    }
    bfs(ultim);
    g<<diametru;
    return 0;
}