Cod sursa(job #1552194)

Utilizator c0mradec0mrade c0mrade Data 17 decembrie 2015 13:13:48
Problema Diametrul unui arbore Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.07 kb
#include<fstream>
#include<vector>
#include<queue>
using namespace std;

ifstream in("darb.in");
ofstream out("darb.out");

vector<int>v[100001];
queue<int>q,q1;
int n,x,y,z,t,sol,a[100001],a1[100001];

void bfs(int root)
{
    a[root]=1;
    q.push(root);
    while(!q.empty())
    {
        t=q.front();
        q.pop();
        for(int i=0;i<v[t].size();++i)
            if(a[v[t][i]]==0)
            {
                a[v[t][i]]=a[t]+1;
                q.push(v[t][i]);
            }
    }
}
void bfs2(int root)
{
    a1[root]=1;
    q1.push(root);
    while(!q1.empty())
    {
        z=q1.front();
        q1.pop();
        for(int i=0;i<v[z].size();++i)
            if(a1[v[z][i]]==0){
                q1.push(v[z][i]);
                a1[v[z][i]]=1;
            }
    }
}

int main(){
    in>>n;
    for(int i=0;i<n-1;++i)
    {
        in>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    bfs2(1);
    bfs(z);
    for(int i=1;i<=n;++i)
        if(a[i]>sol)
            sol=a[i];
    out<<sol;
    return 0;
}