Pagini recente » Cod sursa (job #853510) | Cod sursa (job #1063711) | Cod sursa (job #1969290) | Cod sursa (job #909679) | Cod sursa (job #2195127)
// Componente Tare Conexe (CTC) - Strongly Connected Components (SCC)
// Tarjan - O(n + m)
// https://infoarena.ro/problema/ctc
#include <bits/stdc++.h>
#define NMAX 100010
using namespace std;
class Task {
public:
void solve() {
read_input();
get_result();
print_output();
}
private:
// n = numar de noduri, m = numar de muchii
int n, m;
// adj[node] lista de adiacenta a nodului node pentru graful initial
vector<int> adj[NMAX];
// ctc[i] = componenta tare conexa cu indicele i
vector<vector<int>> ctc;
// in_stack[i] = 1 daca nodul i se afla in stiva
vector<int> in_stack;
// retin nodurile prin care trec
stack<int> st;
vector<int> idx;
vector<int> low_link;
void read_input() {
cin >> 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 = 1; i <= m; ++i) {
cin >> x >> y;
adj[x].push_back(y);
}
}
void get_result() {
tarjan();
}
void tarjan() {
int index = 0;
for (int i = 1; i <= n; ++i) {
if (idx[i] == -1) {
dfs(i, index);
}
}
}
void dfs(int node, int &index) {
idx[node] = low_link[node] = index++;
in_stack[node] = 1;
st.push(node);
for (auto &vecin : adj[node]) {
if (idx[vecin] == -1) {
dfs(vecin, index);
low_link[node] = std::min(low_link[node], low_link[vecin]);
continue;
}
if (in_stack[vecin]) {
low_link[node] = std::min(low_link[node], low_link[vecin]);
}
}
if (low_link[node] == idx[node]) {
vector<int> current_ctc;
int target_node = -1;
while (target_node != node) {
target_node = st.top();
st.pop();
current_ctc.push_back(target_node);
in_stack[target_node] = 0;
}
ctc.push_back(current_ctc);
}
}
void print_output() {
// pentru verificare pe infoarena trebuie doar modificata afisarea
int sol = ctc.size();
cout << sol << '\n';
//cout << "Graful are " << sol << " componente tare conexe\n";
for (int i = 0; i < sol; ++i) {
//cout << "Componenta tare conexa [" << i + 1 << "]: ";
for (auto &node : ctc[i]) {
cout << node << " ";
}
cout << '\n';
}
}
};
int main() {
// din cauza ca fac redirectari, salvez starea lui cin si cout
auto cin_buff = cin.rdbuf();
auto cout_buff = cout.rdbuf();
// las liniile urmatoare daca citesc din fisier
ifstream fin("ctc.in");
cin.rdbuf(fin.rdbuf()); // save and redirect
// las liniile urmatoare daca afisez in fisier
ofstream fout("ctc.out");
cout.rdbuf(fout.rdbuf()); //save and redirect
// aici este rezolvarea propriu-zisa
Task *task = new Task();
task->solve();
delete task;
// restore pentru cin si cout
cin.rdbuf(cin_buff);
cout.rdbuf(cout_buff);
// obs. nu e nevoie sa inchid fisierele
// cand se apeleaza destructorii pentru fin si fout se vor inchide
return 0;
}