Cod sursa(job #2135003)

Utilizator UWantMyNameGheorghe Vlad Camil UWantMyName Data 18 februarie 2018 15:20:51
Problema Cuplaj maxim in graf bipartit Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.12 kb
#include <bits/stdc++.h>
#define in "cuplaj.in"
#define out "cuplaj.out"
#define Nmax 10003
using namespace std;
ifstream fin(in);
ofstream fout(out);

int n,m,E;
int st[Nmax],dr[Nmax];
vector <int> L[Nmax];
int cuplajmax;
int viz[Nmax];

/// st[i] = nodul cu care se cupleaza nodul i din multimea A (cea din stg)
/// dr[i] = nodul cu care se cupleaza nodul i din multimea B (cea din drt)
/// cuplajmax = numarul de muchii din cuplajul maxim
/// viz[i] = 1 daca nodul i din stanga este cuplat


/// aici incerc sa cuplez nodul k
/// return 1 daca pot, 0 daca nu
bool Cupleaza(int k)
{
    if (viz[k] == 1) return false; /// este deja cuplat
    viz[k] = 1; /// il marchez ca este cuplat
    /// acum, cu cine il cuplez?
    for (auto i : L[k])
        if (dr[i] == 0) /// i este necuplat
        {
            st[k] = i;
            dr[i] = k;
            return true;
        }
    /// n-am reusit sa il cuplez pe k
    /// deci, mai parcurg again L[k] si poate reusesc sa cuplez pe k cu un nod cuplat
    for (auto i : L[k])
        if (Cupleaza(dr[i]))
        {
            /// cuplez din nou i cu k, iar pe dr[i] il cuplez cu altul
            st[k] = i;
            dr[i] = k;
            return true;
        }
    return false;
}

void Read()
{
    int x,y,i;

    fin >> n >> m >> E;
    for (i = 1; i <= E; i++)
    {
        fin >> x >> y;
        L[x].push_back(y);
    }
}

void Solve()
{
    int i;
    bool gata = false;

    while(!gata)
    {
        gata = true;

        /// initializez viz-ul
        for (i = 1; i <= n; i++)
            viz[i] = 0;
        for (i = 1; i <= n; i++)
            if (st[i] == 0 && Cupleaza(i))
            {
                /// daca e necuplat si pot sa il cuplez
                cuplajmax++;
                gata = false;
            }
    }
    fout << cuplajmax << "\n";
    for (i = 1; i <= n; i++)
        if (st[i] != 0) fout << i << " " << st[i] << "\n"; /// muchia (i,st[i]); in st[i] retin cu cine este cuplat in dreapta
}

int main()
{
    Read();
    Solve();

    fin.close();
    fout.close();
    return 0;
}