Cod sursa(job #2195050)

Utilizator firutibogdanFiruti Bogdan-Cristian firutibogdan Data 14 aprilie 2018 23:45:40
Problema Componente tare conexe Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.7 kb
// CTC Tarjan

#include <fstream>
#include <iostream>
#include <vector>
#include <stack>

using namespace std;

#define NMAX 100010

class Task {

public:
	void solve() {
		read_input();
		get_result();
		print_output();
	}
private:
	vector<int> adj[NMAX];
	vector<vector<int>> ctc;
	vector<int> idx;
	vector<int> low_link;
	vector<int> in_stack;
	stack<int> st;
	int n, m;

	void read_input() {
		ifstream fin("ctc.in");
		fin >> n >> m;
		in_stack = vector<int> (n + 1, 0);
		idx = vector<int> (n + 1, -1);
		low_link = vector<int> (n + 1, 0);

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

	void get_result() {
		int index = 0;
		for (int i = 1; i <= n; ++i) {
			if (idx[i] == -1) {
				tarjan(i, index);
			}
		}
	}

	void tarjan(int node, int index) {
		idx[node] = index;
		low_link[node] = index;
		in_stack[node] = 1;
		index++;
		st.push(node);

		for (int neighbour : adj[node]) {
			if (idx[neighbour] == -1) {
				tarjan(neighbour, index);
				low_link[node] = min(low_link[node], low_link[neighbour]);
			} else {
				if (in_stack[neighbour] == 1) {
					low_link[node] = min (low_link[node], idx[neighbour]);
				}
			}
		}

		if (low_link[node] == idx[node]) {
			int current;
			vector<int> new_ctc;
			do {
				current = st.top();
				st.pop();
				in_stack[current] = 0;
				new_ctc.push_back(current);
			} while (current != node);
			ctc.push_back(new_ctc);
		}
	}

	void print_output() {
		ofstream fout("ctc.out");
		fout << ctc.size() << "\n";
		for (vector<int> comp : ctc) {
			for (int el : comp) {
				fout << el << " ";
			}
			fout << "\n";
		}
		fout.close();
	}
};

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