Cod sursa(job #2937202)

Utilizator ccazacu1Cazacu Cristian - Gabriel ccazacu1 Data 10 noiembrie 2022 01:09:19
Problema Componente tare conexe Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.11 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <stack>
using namespace std;

ifstream f("ctc.in");
ofstream g("ctc.out");

vector<vector<int>> sol, graph, trans;  // lista de adiacenta a grafului initial si a transpusei pentru Kosaraju
vector<int> viz;        // vector de noduri vizitate
stack<int> st;      // stack -ul pe care incarcam nodurile
int n, m, nrctc;
void dfs(int k)     //parcurgere dfs pe grapful initial
{
    viz[k]=1;       //marcam nodurile parcurse cu 1
    for(auto i: graph[k])
        if(viz[i]==0)
            dfs(i);
    st.push(k);     //incarcam elementele pe stiva
}
void dfstrans(int k)
{
    viz[k]=2;           //marcam varful ca fiind vizitat in transpusa => face parte dintr-o componenta tare conexa
    for(auto i: trans[k])
        if(viz[i]==1)
            dfstrans(i);
    sol[nrctc].push_back(k);        //incarcam nodurile din componenta tare conexa in array-ul specific acesteia
}
int main()
{
    f>>n>>m;    //citim nr de varfuri si muchii
    sol.resize(n + 1);
    viz.resize(n + 1,0);
    graph.resize(n + 1);      // initializam matricele cu n + 1 elemente
    trans.resize(n + 1);
    for(int i=1; i<=m ; i++)
    {
        int x, y;
        f>>x>>y;
        graph[x].push_back(y);      //populam lista de adiacenta a grafului si a transpusei
        trans[y].push_back(x);
    }
    for(int i=1; i<=n; i++)
    {
        if(viz[i]==0)       // pornim un dfs din fiecare nod care nu e vizitat
            dfs(i);
    }
    while(st.size()>0)         //incepem sa golim stack-ul
    {
        if(viz[st.top()]==1)        //verificam daca elementul din varful stivei face parte dintr-o componenta tare conexa ar trebui sa fie marcat cu 2
        {
            nrctc++;            // daca este marcat cu 1 trebuie sa pornim un dfs din el
            dfstrans(st.top());
        }
        st.pop();       // eliminam elementul din stiva dupa dfs sau daca a fost vizitat intr-un dfs al altui nod
    }
    g<<nrctc<<'\n';
    for(int i=1; i<=nrctc; i++)
    {
        for(auto j: sol[i])     //incarcam in fisier componentele tare conexe
            g<<j<<" ";
        g<<'\n';
    }
    return 0;
}