Cod sursa(job #2134673)

Utilizator BourucLiviuBouruc Petru Liviu BourucLiviu Data 18 februarie 2018 11:17:26
Problema Cuplaj maxim in graf bipartit Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.68 kb
#include <fstream>
#include <vector>

#define N 10005

using namespace std;

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

/** st[], dr[] - memoram capetele
    st[i] - nodul i din multimea B
    dr[i] - din mult A
    1. cuplej mereu o muchie din A cu una din b
    2. daca voi cupla un nod i din A atunci i ramane mereu cuplat
**/

int n, m, st[N], dr[N];
vector <int> L[N];
int cuplajMax; /// nr de muchii din cuplajul maxim
int viz[N]; /// viz[i] = 1 daca nodul i din stanga e cuplat

/// incerc sa cuplez nodul k    1 da 0 nu
int Cupleaza(int k)
{
    if(viz[k] == 1) return 0;
    viz[k] = 1;
    for(auto i : L[k])
        if(dr[i] == 0) /// i e necuplat
        {
            st[k] = i;
            dr[i] = k;
            return 1;
        }
    /// n-am reusit sa-l cuplez pe k
    /// mai parcurg odata L[k] si poate reusesc sa cuplez pe k cu un nod cuplat
    for(auto i : L[k]) /// i este din partea dreapta
        if(Cupleaza(dr[i]))
        {
            st[k] = i;
            dr[i] = k;
            return 1;
        }
    return 0;
}

int main()
{
    int e;
    fin >> n >> m >> e;
    for(int i = 1, x, y; i <= e; ++i)
    {
        fin >> x >> y;
        L[x].push_back(y);
    }
    fin.close();

    int ok = 0;
    while(!ok)
    {
        ok = 1;
        for(int i = 1; i <= n; ++i) viz[i] = 0;
        for(int i = 1; i <= n; ++i)
            if(!st[i] && Cupleaza(i))
            {
                cuplajMax++;
                ok = 0;
            }
    }
    fout << cuplajMax << '\n';
    for(int i = 1; i <= n; ++i)
        if(st[i] != 0) fout << i << " " << st[i] << '\n';
    fout.close();
    return 0;
}