Cod sursa(job #1495207)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 2 octombrie 2015 18:54:17
Problema Componente tare conexe Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.75 kb
#include<iostream>
#include<fstream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<bitset>
#include<cstring>
#include<queue>
#include<stack>

#define ull unsigned long long
#define ll long long
#define pb push_back
#define FOR(a,b,c) for (int a=b;a<=c; ++a)
#define ROF(a,b,c) for (int a=b;a>=c; --a)
#define Nmax 100010

// algoritmul plus-minus

using namespace std;
ifstream f("ctc.in");
ofstream g("ctc.out");
int N,M,nrcomp;
vector<int> vorig[Nmax];
vector<int> vtrans[Nmax];
vector<int> componente[Nmax];
bitset<Nmax> verif;
bitset<Nmax> eplus;

void dfs_p(int,vector<int>*);
void dfs_m(int,vector<int>*);

int main()
{
    f>>N>>M;
    FOR (i,1,M) {
        int x,y;
        f>>x>>y;
        vorig[x].pb(y);
        vtrans[y].pb(x);
    }
    FOR (i,1,N) {
        if (!verif.test(i)) {
            ++nrcomp;
            dfs_p(i,vorig);
            dfs_m(i,vtrans);
            FOR (j,1,N) {
                eplus.set(j,0);
            }
        }
    }
    g<<nrcomp<<'\n';
    FOR (i,1,nrcomp) {
        vector<int>::iterator it;
        for (it=componente[i].begin();it<componente[i].end();++it) {
            g<<(*it)<<' ';
        }
        g<<'\n';
    }
    f.close();g.close();
    return 0;
}

void dfs_p(int nod,vector<int> *v) {
    eplus.set(nod,1);
    for (unsigned int k=0;k<v[nod].size();++k) {
        if (!eplus.test(v[nod][k]) && !verif.test(v[nod][k])) {
            dfs_p(v[nod][k],v);
        }
    }
}

void dfs_m(int nod,vector<int> *v) {
    verif.set(nod,1);
    componente[nrcomp].pb(nod);
    for (unsigned int k=0;k<v[nod].size();++k) {
        if (eplus.test(v[nod][k]) && !verif.test(v[nod][k])) {
            dfs_m(v[nod][k],v);
        }
    }
}