Cod sursa(job #2063460)

Utilizator MaligMamaliga cu smantana Malig Data 11 noiembrie 2017 11:42:13
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <iostream>
#include <fstream>
#include <set>
#include <cstring>
#include <cstdlib>
#include <vector>

#if 1
#define pv(x) cout<<#x<<" = "<<x<<"; ";cout.flush()
#define pn cout<<endl
#else
#define pv(x)
#define pn
#endif

using namespace std;
ifstream in("darb.in");
ofstream out("darb.out");

using ll = long long;
using ull = unsigned long long;
using ui = unsigned int;
#define pb push_back
#define mp make_pair
const int NMax = 1e5 + 5;
const int inf = 1e9 + 5;
const int mod = 9973;
using zint = int;

int N,mxNode,mxDist;
vector<int> v[NMax];

void dfs(int,int,int);

int main() {
    in>>N;
    for (int i=1;i < N;++i) {
        int x,y;
        in>>x>>y;

        v[x].pb(y);
        v[y].pb(x);
    }

    mxDist = -1;
    dfs(1,0,1);

    int first = mxNode;
    mxDist = -1;
    dfs(first,0,1);

    out<<mxDist<<'\n';

    in.close();out.close();
    return 0;
}

void dfs(int node,int dad,int dist) {

    for (int nxt : v[node]) {
        if (nxt == dad) {
            continue;
        }

        dfs(nxt,node,dist+1);
    }

    if (mxDist < dist) {
        mxDist = dist;
        mxNode = node;
    }
}