Cod sursa(job #3203180)

Utilizator DomnulMilandruMilandru Nicon-David DomnulMilandru Data 13 februarie 2024 11:08:05
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb

#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream cin("darb.in");
ofstream cout("darb.out");
int n;
int x,y;
vector<vector<int>> A;
vector<int> d;
vector<int> d1;
vector<bool> fr;
int last,maxi;
void dfs(int nod)
{
    if(d1[nod]>maxi)
    {
        maxi=d1[nod];
        last=nod;
    }
    for(int i=0;i<A[nod].size();i++)
    {
        int vecin=A[nod][i];
        if(!fr[vecin])
        {
            fr[vecin]=1;
            d1[vecin]=d1[nod]+1;
            dfs(vecin);
        }
    }
}
void bfs(int nod)
{
    queue<int> Q;
    Q.push(nod);
    while(!Q.empty())
    {
        int cap=Q.front();
        maxi=max(maxi,d[cap]);
        Q.pop();
        for(int i=0;i<A[cap].size();i++)
        {
            int vecin=A[cap][i];
            if(!fr[vecin])
            {
                fr[vecin]=1;
                d[vecin]=d[cap]+1;
                Q.push(vecin);
            }
        }
    }
}
int main()
{
    cin>>n;
    A.resize(n+1);
    d.resize(n+1);
    fr.resize(n+1);
    d1.resize(n+1);
    for(int i=0;i<n-1;i++)
    {
        cin>>x>>y;
        A[x].push_back(y);
        A[y].push_back(x);
    }
    fr[1]=1;
    d1[1]=1;
    dfs(1);
    fr.resize(0);
    fr.resize(n+1);
    fr[last]=1;
    d[last]=1;
    maxi=0;
    bfs(last);
    cout<<maxi;
    return 0;
}