Cod sursa(job #3228945)

Utilizator Ionut2212Nedelcu Alexandru Ionut Ionut2212 Data 12 mai 2024 15:38:32
Problema Diametrul unui arbore Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.28 kb
#include <iostream>
#include <vector>
#include <fstream>

using namespace std;
ifstream fin ("darb.in");
ofstream fout("darb.out");
int dmax, nodmax;
int viz2[200001];
int viz[200001];
vector<vector<int> > v;
void dfs(int nod)
{
    for(int i = 0; i < v[nod].size(); i++)
    {
        int vecin = v[nod][i];
        if(!viz[vecin])
        {
            viz[vecin] = viz[nod] + 1;
            if(viz[vecin] > dmax)
            {
                dmax = viz[vecin];
                nodmax = vecin;
            }
            dfs(vecin);
        }
    }
}
void dfs2(int nod)
{
    for(int i = 0; i<v[nod].size(); i++)
    {
        int vecin = v[nod][i];
        if(!viz2[vecin])
        {
            viz2[vecin] = viz2[nod] + 1;
            if(viz2[vecin] > dmax)
            {
                dmax = viz2[vecin];
                nodmax = vecin;
            }
            dfs2(vecin);
        }
    }
}
int main()
{
    int n;
    cin >> n;
    v.resize(n+1);
    for(int i = 1; i<n; i++)
    {
        int a,b;
        cin >> a>> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    viz[1] = 1;
    dfs(1);
    viz2[nodmax] = 1;
    dmax = 0;
    dfs2(nodmax);
    cout << dmax;
    return 0;
}
/*
6
1 2
1 3
3 4
3 5
2 6
*/