Cod sursa(job #2035981)

Utilizator MaligMamaliga cu smantana Malig Data 10 octombrie 2017 08:36:11
Problema Componente tare conexe Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.51 kb
#include <iostream>
#include <fstream>
#include <set>
#include <vector>
#include <stack>

#define ll long long
#define pb push_back
using namespace std;
ifstream in("ctc.in");
ofstream out("ctc.out");

const int NMax = 1e5 + 5;
const int arbMax = 262144 + 5;

int N,M,nrComp;
bool inComp[NMax],visStack[NMax];
vector<int> v[NMax],rev[NMax],comp[NMax];
stack<int> st;

void buildStack(int);
void buildComp(int);

int main() {
    in>>N>>M;

    while (M--) {
        int x,y;
        in>>x>>y;

        v[x].pb(y);
        rev[y].pb(x);
    }

    for (int i=1;i <= N;++i) {
        if (visStack[i]) {
            continue;
        }

        buildStack(i);
    }

    while (st.size()) {
        int node = st.top(); st.pop();
        if (inComp[node] != 0) {
            continue;
        }

        ++nrComp;
        buildComp(node);
    }

    out<<nrComp<<'\n';
    for (int i=1;i <= nrComp;++i) {
        for (int node : comp[i]) {
            out<<node<<' ';
        }
        out<<'\n';
    }

    in.close();out.close();
    return 0;
}

void buildStack(int node) {
    visStack[node] = true;

    for (int nxt : v[node]) {
        if (visStack[nxt]) {
            continue;
        }

        buildStack(nxt);
    }

    st.push(node);
}

void buildComp(int node) {
    comp[nrComp].pb(node);
    inComp[node] = true;

    for (int nxt : rev[node]) {
        if (inComp[nxt]) {
            continue;
        }

        buildComp(nxt);
    }
}