Cod sursa(job #1875289)

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

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

vector<int> v[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])
    {
        if(dr[it] == 0)
        {
            if(dr[it] == 0) /// daca dreapta nu a fost conectata
            {
                cout << "\t-vecin lui "<<nod<<": "<<it<<'\n';
                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);
    }
    ///initializam conectiunea triviala
    for(int i=1; i<=n; i++)
    {
        ok = 0;
        for(auto it:v[i])
        {
            if(dr[it] == 0)
            {
                dr[it] = i;
                st[i] = it;
                muchie++;
                ok = 1;
                break;
            }
        }
        if(!ok)
            necupl.push_back(i);
    }
    /// 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;
}