Cod sursa(job #3122416)

Utilizator unomMirel Costel unom Data 18 aprilie 2023 22:35:01
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.95 kb
#include <fstream>
#include <vector>

using namespace std;

ifstream in("darb.in");
ofstream out("darb.out");

struct node
{
    vector<int> vec;
};
int n;
int viz[100005];
node v[100005];

void dfs(int k, int val)
{
    viz[k] = val;

    for(int i = 0; i<v[k].vec.size(); i++)
    {
        if(viz[v[k].vec[i]] == 0)
        {
            dfs(v[k].vec[i], val+1);
        }
    }
}

int main()
{
    in>>n;
    int st, dr;
    for(int i = 1; i<n; i++)
    {
        in>>st>>dr;
        v[st].vec.push_back(dr);
        v[dr].vec.push_back(st);
    }

    dfs(1, 1);

    int dmax = 0;
    int poz = 0;

    for(int i = 1; i<=n; i++)
    {
        if(viz[i] > dmax)
        {
            dmax = viz[i];
            poz = i;
        }
        viz[i] = 0;
    }

    dfs(poz, 1);

    dmax = 0;
    for(int i = 1; i<=n; i++)
    {
        dmax = max(dmax, viz[i]);
    }

    out<<dmax;

    return 0;
}