Cod sursa(job #2749436)

Utilizator ioana0211Ioana Popa ioana0211 Data 6 mai 2021 17:56:12
Problema Diametrul unui arbore Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.86 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>

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

const int MAXN=100005;
int n, x, y;
vector<int> g[MAXN];
queue<int>q;
int viz[MAXN], frunza;

void bfs(int start_node)
{
    memset(viz, 0, MAXN);
    q.push(start_node);
    viz[start_node]=1;
    frunza=0;
    while(!q.empty())
    {
        int crt_node=q.front();
        for(auto &v: g[crt_node])
            if(!viz[v]){
                q.push(v);
                viz[v]=viz[crt_node]+1;
            }
        q.pop();
        frunza=crt_node;
    }
}
int main()
{
    fin>>n;
    for(int i=1; i<=n-1; i++)
    {
        fin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    bfs(1);
    bfs(frunza);
    fout<<viz[frunza];
    return 0;
}