Cod sursa(job #667180)

Utilizator Bit_MasterAlexandru-Iancu Caragicu Bit_Master Data 22 ianuarie 2012 18:27:34
Problema Cuplaj maxim in graf bipartit Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.96 kb
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

const int N_MAX = 10010;
const int M_MAX = 10010;

int n,m;
vector<int> vecin[N_MAX]; //vecinul din multimea a doua
bool vizitat[N_MAX];
int l_cuplaj = 0;
bool cuplat_a[N_MAX]; //daca este cuplat nodul din multimea 1
int cuplat_b[M_MAX]; //nodul din multimea 1 cu care este cuplat acum nodul din multimea 2

void citeste()
{
    int e,a,b;
    freopen("cuplaj.in","r",stdin);
    freopen("cuplaj.out","w",stdout);
    scanf("%d%d%d",&n,&m,&e);
    for (int i = 1; i <= e; ++i)
    {
        scanf("%d%d",&a,&b);
        vecin[a].push_back(b);
    }
}

int cuplare(int nod)  //intoarce 1 daca s-a putut cupla nodul i sau recupla daca era daja cuplat, 0 altfel
{
    int dest;
    if (vizitat[nod])
        return 0;
    vizitat[nod] = true;
    for (unsigned int i = 0; i < vecin[nod].size(); ++i)
    {
        dest = vecin[nod][i];
        if (cuplat_b[dest] == 0 || cuplare(cuplat_b[dest]))
        {
            //cuplat_a[cuplat_b[dest]] = false;
            cuplat_a[nod] = true;
            cuplat_b[dest] = nod;
            return 1;
        }
    }
    return 0;
}

void cupleaza()
{
    /**
     * while (true) {
     *   memset(vizitat, 0, sizeof(vizitat));
     *   for i = 1; i <= n; ++i
     *     if i nu apartine cuplajului
     *       cuplare(i);
     * }
     */
    bool continuare = true;
    while (continuare)
    {
        continuare = false;
        memset(vizitat,0,sizeof(vizitat));
        for (int i = 1; i <= n; ++i)
            if (!cuplat_a[i])
                if (cuplare(i))
                {
                    continuare = true;
                    ++l_cuplaj;
                }
    }
}

void scrie()
{
    printf("%d\n",l_cuplaj);
    for (int i = 1; i <= m; ++i)
        if (cuplat_b[i] != 0)
            printf("%d %d\n",cuplat_b[i],i);
}

int main()
{
    citeste();
    cupleaza();
    scrie();
}