Cod sursa(job #2812327)

Utilizator mirunavrAvram Miruna-Alexandra mirunavr Data 4 decembrie 2021 13:22:41
Problema Diametrul unui arbore Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.94 kb
#include<bits/stdc++.h>
using namespace std;


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


vector<int> v[100001];
int vis[100001];
queue<int> q;
int cost[100001];
int n,x,y;
void add(int node1,int node2){
    v[node1].push_back(node2);
    v[node2].push_back(node1);
}

void bfs(int node,int &last){
    int a;
    for(int i=1;i<=n;i++)
        vis[i]=0;
    for(int i=1;i<=n;i++)
        cost[i]=-1;
    vis[node]=1;
    cost[node]=0;
    q.push(node);
    while(!q.empty()){
        node=q.front();
        //g<<node<<" ";
        last=node;
        a=cost[node];
        q.pop();
        for(auto i:v[node])
        {
            if(vis[i]==0){
                vis[i]=1;
                cost[i]=a+1;
                q.push(i);
            }
        }
    }

}
int main(){
    f>>n;
    int first_node,final;
    for(int i=0;i<n-1;i++){
        f>>x>>y;
        add(x,y);
    }
    bfs(1,first_node);
    bfs(first_node,final);
    g<<cost[final]+1;

}