Cod sursa(job #2610301)

Utilizator mzzmaraMara-Ioana Nicolae mzzmara Data 4 mai 2020 18:26:51
Problema Diametrul unui arbore Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.13 kb
#include <bits/stdc++.h> 

using namespace std;

const int kNmax = 100005;

class Task {
 public:
	void solve() {
		read_input();
        get_result();
		print_output();
	}

 private:
	int n;
	vector<int> adj[kNmax];
    int maxSize;

	void read_input() {
		ifstream fin("darb.in");

		fin >> n;
	
		for (int i = 1, x, y; i < n; i++) {
			fin >> x >> y;
			adj[x].push_back(y);
            adj[y].push_back(x);
		}
		fin.close();
	}


	void get_result() {
        maxSize = -1;
		std::vector<bool> visited(n + 1, false); 

        for (int i = 1; i <= n; i++) {
            dfsHelper(i, visited, 1);
		}
    }

    void dfsHelper(int node, std::vector<bool> &visited, int size) { 
        maxSize = std::max(maxSize, size);
        visited[node] = true;

        for (int i = 0; i < adj[node].size(); i++) { 
            if (!visited[adj[node][i]]) { 
                dfsHelper(adj[node][i], visited, size + 1);
            }
        }

		visited[node] = false;
    }

	void print_output() {
		ofstream fout("darb.out");
		fout << maxSize << "\n";
		fout.close();
	}
};

int main() {
	Task *task = new Task();
	task->solve();
	delete task;
	return 0;
}