Cod sursa(job #2108024)

Utilizator Laura_CorneiLaura Maria Cornei Laura_Cornei Data 17 ianuarie 2018 20:55:31
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.05 kb
#include <fstream>
#include <vector>
#define nmax 100005
using namespace std;
fstream f1("darb.in", ios::in);
fstream f2("darb.out", ios::out);
int n, nod1, viz1[nmax], viz2[nmax], maxi1, maxi2, nod2;
vector <int> v[nmax];
void dfs1(int nod)
{
    for(auto it=v[nod].begin(); it!=v[nod].end(); ++it)
        if(!viz1[*it])
          {
              viz1[*(it)]=viz1[nod]+1;
              if(maxi1< (viz1[*(it)])) {nod1=*(it); maxi1=viz1[*(it)];}
              dfs1(*it);
          }
}
void dfs2(int nod)
{
    for(auto it=v[nod].begin(); it!=v[nod].end(); ++it)
        if(!viz2[*it])
          {
              viz2[*(it)]=viz2[nod]+1;
              if(maxi2< (viz2[*(it)])) {nod2=*(it); maxi2=viz2[*(it)];}
              dfs2(*it);
          }
}
int main()
{
    int i, x, y;
    f1>>n;
    for(i=1; i<=n; i++)
    {
        f1>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    maxi1=1;
    nod1=1;
    viz1[1]=1;
    dfs1(1);

    viz2[nod1]=1;
    maxi2=1;
    dfs2(nod1);
    f2<<maxi2;
    return 0;
}