Cod sursa(job #2093640)

Utilizator MaligMamaliga cu smantana Malig Data 24 decembrie 2017 12:08:33
Problema Componente tare conexe Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.4 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <stack>

using namespace std;
ifstream in("ctc.in");
ofstream out("ctc.out");

#define pb push_back
typedef long long ll;
const int NMax = 1e5 + 5;

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

void dfsPlus(int);
void dfsMinus(int);

int main() {
    in>>N>>M;
    for (int i=1;i <= M;++i) {
        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;

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

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

        dfsPlus(nxt);
    }
}

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

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

        dfsMinus(nxt);
    }
}