Cod sursa(job #2195127)

Utilizator pakistanezuPopescu Alexandru Gabriel pakistanezu Data 15 aprilie 2018 13:31:53
Problema Componente tare conexe Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 3.3 kb
// 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;
}