Cod sursa(job #2915237)

Utilizator bumblebeeGeorge Bondar bumblebee Data 22 iulie 2022 09:40:24
Problema Diametrul unui arbore Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>
#include <vector>
#include <cstring>
#define MAX_NODES 100001
using namespace std;

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

vector<int> tree[MAX_NODES];
int distances[MAX_NODES], visited[MAX_NODES], diameter;

void compute_distance(int current) {
    visited[current] = 1;
    for (int &adjacent_node : tree[current]) {
        if (!visited[adjacent_node]) {
            distances[adjacent_node] = distances[current] + 1;
            if (distances[adjacent_node] > diameter) {
                diameter = distances[adjacent_node];
            }
            compute_distance(adjacent_node);
        }
    }
}

int findDiameter(int nr_nodes) {
    for (int i = 1; i <= nr_nodes; ++i) {
        compute_distance(i);
        memset(distances, 0, sizeof(distances));
        memset(visited, 0, sizeof(distances));
    }
    return diameter + 1;
}

int main() {
    int nr_nodes, node_a, node_b;
    fin >> nr_nodes;
    for (int i = 0; i < nr_nodes; ++i) {
        fin >> node_a >> node_b;
        tree[node_a].push_back(node_b);
        tree[node_b].push_back(node_a);
    }
    fout << findDiameter(nr_nodes);
    return 0;
}