Cod sursa(job #2008662)

Utilizator MaligMamaliga cu smantana Malig Data 7 august 2017 11:19:34
Problema Componente tare conexe Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.46 kb
#include <iostream>
#include <fstream>
#include <unordered_map>
#include <queue>
#include <stack>
#include <vector>

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

const int NMax = 1e5 + 5;
const ll inf = 1e18 + 5;

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

void dfs1(int);
void dfs2(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 (mns[i]) {
            continue;
        }

        ++nrComp;
        dfs1(i);
        dfs2(i);

        while (st.size()) {
            pls[st.top()] = false;
            st.pop();
        }
    }

    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 dfs1(int node) {
    pls[node] = true;
    st.push(node);

    for (int nxt : v[node]) {
        if (pls[nxt] || mns[nxt]) {
            continue;
        }

        dfs1(nxt);
    }
}

void dfs2(int node) {
    mns[node] = true;
    comp[nrComp].pb(node);

    for (int nxt : rev[node]) {
        if (!pls[nxt] || mns[nxt]) {
            continue;
        }

        dfs2(nxt);
    }
}