Cod sursa(job #2480813)

Utilizator alexmisto342Turdean Alexandru alexmisto342 Data 26 octombrie 2019 10:31:20
Problema Diametrul unui arbore Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.27 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("darb.in");
ofstream fout("darb.out");
vector<int> vecini[100005];

int raspuns[100005];
int lant[1000005];
int n;

void dfs(int nod_curent, int parinte)
{
    if(raspuns[nod_curent] != 0)  /// daca am mai vizitata nodul oprim functia
        return;
    raspuns[nod_curent] = -1;
    int maxi1 = 0, maxi2 = 0;
    for(int i = 0; i < vecini[nod_curent].size(); i++)
    {
        int nod_urmator = vecini[nod_curent][i];
        if(nod_urmator != parinte)
        {
            dfs(nod_urmator, nod_curent);
            lant[nod_curent] = max(lant[nod_curent], lant[nod_urmator] + 1);
            if(raspuns[nod_urmator] > maxi1)
            {
                maxi2 = maxi1;
                maxi1 = raspuns[nod_urmator];
            }
            else if(raspuns[nod_urmator] > maxi2)
            {
                maxi2 = raspuns[nod_urmator];
            }
        }
    }
    raspuns[nod_curent] = max(lant[nod_curent], maxi1 + maxi2 + 1);
}

int main()
{
    fin >> n;
    for(int i = 1; i <= n; i++)
    {
        int a,b;
        fin >> a >> b;
        vecini[a].push_back(b);
        vecini[b].push_back(a);
    }

    dfs(1, -1);
    fout << raspuns[1];

    return 0;
}