Cod sursa(job #2695612)

Utilizator stanbianca611Stan Bianca stanbianca611 Data 13 ianuarie 2021 22:08:05
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.14 kb
#include <iostream>
#include <fstream>
#include <bits/stdc++.h>
using namespace std;
ifstream f ("darb.in");
ofstream g ("darb.out");
int d_in[100005], d_out[100005];

vector<int> v[100005];
queue<int> q;
vector<int> viz(100005);
int ultim;

void bfs()
{
    while(!q.empty())
    {
        int x=q.front();
        ultim=x;
        q.pop();
        for(int i=0; i<v[x].size(); i++)
        {
            if(viz[v[x][i]]==-1)
            {
                q.push(v[x][i]);
                viz[v[x][i]]=viz[x]+1;
            }
        }
    }
}

int main()
{
    int n, s;
    f>>n;
    for(int i=1; i<=n-1; i++)
    {
        int x, y;
        f>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
        d_in[y]++;
        d_out[x]++;
    }
    for(int i=1; i<=n; i++)
    {
        if(d_in[i]==0) s=i;
    }
    fill(viz.begin(), viz.end(), -1);
    viz[s]=0;
    q.push(s);
    bfs();
    fill(viz.begin(), viz.end(), -1);
    q.push(ultim);
    viz[ultim]=0;
    bfs();
    int maxx=0;
    for(int i=1; i<=n; i++)
    {
        maxx=max(maxx, viz[i]);
    }
    g<<maxx+1;
    return 0;
}