Cod sursa(job #2570271)

Utilizator cyg_Alex_codegicianBarbu Alexandru cyg_Alex_codegician Data 4 martie 2020 15:50:50
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda r3capitusulare Marime 1 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin ("darb.in");
ofstream fout ("darb.out");
const int nmax=100005;
int n,grad[nmax],dist[nmax],rez;
bool viz[nmax];
vector <int> muchii[nmax];
void dfs(int nod)
{
    viz[nod]=true;
    for (int i=0;i<muchii[nod].size();i++)
    {
        int vecin=muchii[nod][i];
        if (!viz[vecin])
        {
            dist[vecin]=dist[nod]+1;
            dfs(vecin);
        }
    }
}
int main()
{
    fin >> n;
    for (int i=1;i<n;i++)
    {
        int x,y;
        fin >> x >> y;
        muchii[x].push_back(y);
        muchii[y].push_back(x);
        grad[x]++;
    }
    dist[3]=1;
    dfs(3);
    int maxx=0,fr=1;
    for (int i=1;i<=n;i++)
    {
        if (dist[i]>maxx)
        {
            fr=i;
            maxx=dist[i];
        }
    }
    maxx=0;
    for (int i=1;i<=n;i++)
    {
        viz[i]=false;
        dist[i]=0;
    }
    dist[fr]=1;
    dfs(fr);
    for (int i=1;i<=n;i++)
    {
        maxx=max(dist[i],maxx);
    }
    fout << maxx;
}