Cod sursa(job #1875299)

Utilizator vladbatalanBatalan Vlad vladbatalan Data 10 februarie 2017 22:34:58
Problema Cuplaj maxim in graf bipartit Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 3.19 kb
#include <bits/stdc++.h>
using namespace std;

ifstream fin("cuplaj.in");
ofstream fout("cuplaj.out");

vector<int> v[10010], pred[10010];
vector<int> necupl;
int st[10010], dr[10010];
int n, m, k, x, y, muchie;
int test_pasi;
bitset<10010> viz;
bool ok;

bool creste(int nod)
{
    //cout << "Verificam daca se poate creste "<<nod<<" \n";
    //test_pasi ++; cout<<"NR. PAS: "<<test_pasi<<'\n';
    viz[nod] = 1;
    //cout << "\t -viz: "; for(int i=1; i<=n; i++) cout<<viz[i]<<' '; cout<<'\n';
    for(auto it:v[nod])
    {
        //cout << "\t-vecin lui "<<nod<<": "<<it<<'\n';
        if(dr[it] == 0) /// daca dreapta nu a fost conectata
        {
            //cout<<"\t\tnu este conectat, conectam "<<nod<<" cu "<<it<<'\n';
            ///realizam conectiune
            st[nod] = it;
            dr[it] = nod;
            viz[nod] = 0;
            return 1;
        }
    }
    for(auto it:v[nod])
    {
        //cout << "\t-vecin lui "<<nod<<": "<<it<<'\n';
        if(dr[it] == 0) /// daca dreapta nu a fost conectata
        {
            //cout<<"\t\tnu este conectat, conectam "<<nod<<" cu "<<it<<'\n';
            ///realizam conectiune
            st[nod] = it;
            dr[it] = nod;
            viz[nod] = 0;
            return 1;
        }
        else
        {
//            cout<< "\t\teste conectat ";
            ///dreapta a fost conectata
            if(!viz[dr[it]]) /// daca dreapta de it nu a fost vizitata
            {
//                cout<<" si nu este vizitat, deci dac il putem creste stangul, adica "<<st[it]<<"\n";
                if(creste(dr[it])) /// daca muchiile pot fi crescute
                {
//                    cout<<"\tne intoarcem in vecinul lui "<<nod<<", in "<<it<<" si le cuplam\n";
                    st[nod] = it;
                    dr[it] = nod;
                    return 1;
                }
            }
//            else
//            {
////                cout<<" este vizitat deja, asa ca vom cauta alta ruta ... \n";
//            }
        }
    }
    viz[nod] = 0;
    return 0;
}

int main()
{
    fin >> n >> m;
    fin >> k;
    for(int i=1; i<=k; i++)
    {
        fin >> x >> y;
        v[x].push_back(y);
        pred[y].push_back(x);
    }
//
//    for(int i=1; i<=n; i++)
//    {
//        fout<<i<<": ";
//        for(auto it:v[i])
//            fout<<it<<' ';
//        fout<<'\n';
//    }
    ///initializam conectiunea triviala
    for(int i=1; i<=n; i++)
    {
        ok = 0;
        for(auto it:v[i])
        {
            if(st[it] == 0)
            {
                st[it] = i;
                dr[i] = it;
                muchie++;
                ok = 1;
                break;
            }
        }
        if(!ok){

            necupl.push_back(i);
        }
    }
//    for(int i=1; i<=n;i++)
//    {
//        fout<< st[i] <<' '<<dr[i]<<'\n';
//    }
    /// mergem pe toti membrii stg, daca un membru nu este conectat se apeleaza functia creste
    for(auto it:necupl)
        if(creste(it))
            muchie ++;

    fout<<muchie<<'\n';
    for(int i=1; i<=n; i++)
        if(st[i]!=0)
            fout<<i<<' '<<st[i]<<'\n';

    return 0;
}