Cod sursa(job #2542536)

Utilizator Giulian617Buzatu Giulian Giulian617 Data 10 februarie 2020 09:27:02
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.89 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("darb.in");
ofstream g("darb.out");
const int NMAX=1000005;
int n,x,y,last,diametru,viz[NMAX];
vector<int>v[NMAX];
queue<int>q;
void bfs(int x)
{
    q.push(x);
    viz[x]=1;
    while(!q.empty())
    {
        x=q.front();
        q.pop();
        for(int i=0; i<v[x].size(); i++)
            if(viz[v[x][i]]==0)
            {
                q.push(v[x][i]);
                viz[v[x][i]]=viz[x]+1;
                if(viz[v[x][i]]>diametru)
                {
                    diametru=viz[v[x][i]];
                    last=v[x][i];
                }
            }
    }
}
int main()
{
    f>>n;
    for(int i=1; i<n; i++)
    {
        f>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    bfs(1);
    for(int i=1;i<=n;i++)
        viz[i]=0;
    bfs(last);
    g<<diametru;
    return 0;
}