Cod sursa(job #2741512)

Utilizator MateGMGozner Mate MateGM Data 16 aprilie 2021 11:48:58
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.9 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <stack>
#include <queue>
using namespace std;

int dim;

int bfs(const vector<vector<int> >&adj,int n,int start)
{
    vector<int>tav(n+1,-1);
    queue<int>q;
    q.push(start);
    tav[start]=1;
    while(!q.empty()){
        int v=q.front();
        q.pop();
        for(int sz:adj[v])
        {
            if(tav[sz]==-1)
            {
                tav[sz]=tav[v]+1;
                dim=max(dim,tav[sz]);
                q.push(sz);
            }
        }
        if(q.empty())
            return v;

    }
}
int main()
{
    ifstream be("darb.in");
    ofstream ki("darb.out");
    int n;
    be>>n;
    vector<vector<int> >adj(n+1);
    for(int i=1;i<=n;i++)
    {
        int x,y;
        be>>x>>y;
        adj[x].push_back(y);
        adj[y].push_back(x);
    }
    int x=bfs(adj,n,1);
    x=bfs(adj,n,x);
    ki<<dim<<endl;
    return 0;
}