Cod sursa(job #1999309)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 10 iulie 2017 21:17:50
Problema Componente tare conexe Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 2.62 kb
#include <iostream>
#include <fstream>
#include <stack>
#include <vector>

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

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

int N,M,nrComp;
bool pls[NMax],mns[NMax];
vector<int> v[NMax],conv[NMax],comp[NMax];
stack<int> st;
// nrComp - numarul de componente gasite pana la un moment dat;
// pls[i] = true daca nodul i a fost marcat cu plus in parcurgerea curenta;
// mns[i] = true daca nodul i a fost marcat cu minus la o parcurgere precedenta
//          si apartine astfel unei componente tare conexa;
// v - listele de adiacenta ale grafului;
// conv - listele de adiacenta ale grafului transpus;
// comp[i] - lista a nodurilor din a i-a componenta gasita;
// st - stiva ce retine nodurile pentru care va fi necesara resetarea vectorului pls,
//      pentru eficienta;

void dfs1(int);
void dfs2(int);

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

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

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

        ++nrComp;
        dfs1(i); // se marcheaza nodurile spre care poate ajunge i cu plus

        // se face un dfs din i in graful transpus, obtinandu-se astfel
        // nodurile din care se poate ajunge in i
        dfs2(i); // se marcheaza nodurile dinspre care se poate ajunge la i cu minus

        // o idee de optimizare este aceea ca la dfs2
        // nu se mai parcurg toate nodurile x,
        // anterioare nodului y (exista arcul (x,y)),
        // care n-au fost marcate si cu plus, deoarece
        // se poate arata ca daca nodul x face parte din componenta lui i
        // atunci exista un drum de la i la x si deci nodul ar fi fost marcat cu plus

        // se reseteaza pls
        while (st.size()) {
            int node = st.top();
            st.pop();

            pls[node] = false;
        }
    }

    out<<nrComp<<'\n';
    for (int k=1;k <= nrComp;++k) {
        for (int node : comp[k]) {
            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 : conv[node]) {
        if (!pls[nxt] || mns[nxt]) {
            continue;
        }

        dfs2(nxt);
    }
}