Cod sursa(job #2928724)

Utilizator alexdn6Dina Alexandru alexdn6 Data 23 octombrie 2022 18:55:46
Problema Componente tare conexe Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.94 kb
// Pentru rezolvare am folosit algoritmul lui Tarjan, la care am adăugat un contor pentru a reține numărul componentelor tare conexe

#include <iostream>
#include <vector>
#include <fstream>
#include <stack>
using namespace std;
ifstream f ("ctc.in");
ofstream g ("ctc.out");
stack <int> stiva;
vector <bool> peStiva;
vector <int> lowlink, index;
int timp = 0, nrCtc = 0;
vector <vector<int>> lista_ad, componente;
void tareConexe(int nod) {

    index[nod] = timp;
    lowlink[nod] = timp;
    timp += 1;
    stiva.push(nod);
    peStiva[nod] = true;

    for(auto n: lista_ad[nod]) {
        if (index[n] == -1) {  // dacă n nu e vizitat
            tareConexe(n);
            lowlink[nod] = min(lowlink[nod], lowlink[n]);
        }
        else if(peStiva[n])
            lowlink[nod] = min(lowlink[nod], index[n]);
    }
    if(lowlink[nod] == index[nod]) {  // dacă nod este nodul "cap" al CTC

        while(stiva.top() != nod) {
            peStiva[stiva.top()] = false;
            componente[nrCtc].push_back(stiva.top());         // reținem nodurile corespunzătoare componentei cu indicele nrCtc
            stiva.pop();
        }
        componente[nrCtc].push_back(stiva.top());
        nrCtc++;			// creștem contorul
        peStiva[stiva.top()] = false;
        stiva.pop();
    }
}
int main() {
    int n, m;
    f >> n >> m;
    lista_ad.resize(n + 1);
    index.resize(n + 1, -1);
    lowlink.resize(n + 1, -1);
    peStiva.resize(n + 1, false);
    componente.resize(m);
    for(int i = 1; i <= m; i++) {
        int a, b;
        f >> a >> b;
        lista_ad[a].push_back(b);
    }

    for(int i = 1; i <= n; i++)
        if(index[i] == -1) {
            tareConexe(i);
        }

    g << nrCtc << "\n";

    for(int i = 0; i < nrCtc; i++) {
        for(int j = 0; j < componente[i].size(); j++)          // afișăm nodurile corespunzătoare pentru fiecare componentă
            g << componente[i][j] << " ";
        g << "\n";
    }
    return 0;
}