Cod sursa(job #3293712)

Utilizator robert111Peter Robert robert111 Data 12 aprilie 2025 13:33:02
Problema Diametrul unui arbore Scor 80
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.14 kb
#include <fstream>
#include <vector>

using namespace std;
ifstream cin("darb.in");
ofstream cout("darb.out");
vector<int>v[100005];
int dfs(int nd,int lst)
{
    int ad=1,mx1=0;
    for(auto i:v[nd])
    {
        if(i!=lst)
            mx1=max(mx1,dfs(i,nd));
    }
    return ad+mx1;
}
int main()
{
    int n,a,b;
    cin>>n;
    for(int i=1;i<n;i++)
    {
        cin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    int nd=1,lst=1;
    if(v[1].size()==1)
    {
        nd=v[1][0];
        while(v[nd].size()==2)
        {
            if(v[nd][0]!=lst)
            {
                lst=nd;
                nd=v[nd][0];
            }
            else if(v[nd][1]!=lst)
            {
                lst=nd;
                nd=v[nd][1];
            }
        }
    }
    int mx1=0,mx2=0;
    for(auto i:v[nd])
    {
        if(i!=lst)
        {
            int nr=dfs(i,nd);
            if(nr>mx1)
            {
                mx2=mx1;
                mx1=nr;
            }
            else if(nr>mx2)
                mx2=nr;
        }
    }
    cout<<mx1+mx2+1;
    return 0;
}