Cod sursa(job #1228347)

Utilizator claudiumihailClaudiu Mihail claudiumihail Data 13 septembrie 2014 21:42:55
Problema Diametrul unui arbore Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.48 kb
#include <fstream>
#include <iostream>
#include <vector>
#include <iterator>
//#include <algorithm>

#define MAXN 100002

using namespace std;

typedef vector<vector<int>> Graph;

Graph graph;

bool visited[MAXN];

int dfs(int node)
{
    int dist = 0;
    for (int neighbor : graph[node])
    {
        if (visited[neighbor] == false)
        {
            visited[neighbor] = true;
            dist = max(dist, dfs(neighbor)) + 1;
        }
    }
    
    return dist;
}

int main()
{
    int n;
    fstream fin("darb.in", fstream::in);
    fstream fout("darb.out", fstream::out);

    fin >> n;
    //cout << n << endl;
    
    graph.resize(n);

    for (int i=0; i<n-1; ++i)
    {
        int x, y;
        fin >> x >> y;
        //cout << x << " " << y << endl;
        
        graph[x-1].push_back(y-1);
        graph[y-1].push_back(x-1);
    }
    
    /*for (int i=0; i<n; ++i)
    {
        cout << i+1 << " ---> ";
        for (int node : graph[i])
        {
            cout << node+1 << " ";
        }
        cout << endl;
    }*/
    
    int mins[2] = {};
    
    visited[0] = true;
    for (int node : graph[0])
    {
        int dist = dfs(node);
        
        if (dist >= mins[0])
        {
            mins[1] = mins[0];
            mins[0] = dist;
        }
        else if (dist > mins[1])
        {
            mins[1] = dist;
        }
    }
    
    fout << mins[0] + mins[1] + 1 << "\n";

    return 0;
}