Cod sursa(job #2957509)

Utilizator mirceaspPetcu Mircea mirceasp Data 22 decembrie 2022 19:02:41
Problema Cuplaj maxim in graf bipartit Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.74 kb
#include <iostream>
#include <bits/stdc++.h>
#include <fstream>

#define mx 10001
using namespace std;
ifstream f("cuplaj.in");
ofstream g("cuplaj.out");
#define minim min(x,flux)
int n, m, x, y, capacitate, e;
int rgraf[2 * mx][2 * mx];

bool bfs(int s, int d, int tata[], bool vizitat[]) {
    queue<int> q;
    vizitat[s] = true;
    tata[s] = -1;
    q.push(s);
    while (!q.empty()) {
        int front = q.front();
        q.pop();
        for (int i = 1; i <= n + m + 1; ++i) {
            if (vizitat[i] == false && rgraf[front][i] > 0) {
                if (i == n + m + 1) {
                    tata[i] = front;
                    return true;
                }
                vizitat[i] = true;
                q.push(i);
                tata[i] = front;
            }
        }
    }
    //daca se poate ajunge la destinatie
    return vizitat[n + m + 1];
}

int main() {
    f >> n >> m >> e;
    for (int i = 1; i <= n; ++i) {
        rgraf[0][i] = 1;
    }
    for (int i = n + 1; i <= m + n; ++i) {
        rgraf[i][m + n + 1] = 1;
    }

    for (int i = 0; i < e; i++) {
        f >> x >> y;
        rgraf[x][y + n] = 1;
    }

    int tata[n + m + 2];
    for (int i = 0; i <= m + n + 1; i++)
        tata[i] = -1;

    bool vizitat[n + m + 2];
    for (int i = 0; i <= n + m + 1; i++)
        vizitat[i] = false;
    int cuplaje = 0;

    vector<int> veciniFinal;
    for (int i = n + 1; i <= m + n; i++)
        veciniFinal.push_back(i);
    while (bfs(0, n + m + 1, tata, vizitat)) {
        for (auto j: veciniFinal) {
            if (rgraf[j][n + m + 1] > 0 && vizitat[j] == true) {
                tata[n + m + 1] = j;

                int flux = INT_MAX;
                int u = n + m + 1;
                while (u != 0) {
                    int x = rgraf[tata[u]][u];
                    flux = minim;
                    u = tata[u];
                }

                if (flux != 0) {
                    cuplaje += 1;
                    u = n + m + 1;
                    while (u != 0) {
                        int vecin = tata[u];
                        rgraf[vecin][u] -= flux;
                        rgraf[u][vecin] += flux;
                        u = tata[u];
                    }
                }

            }
        }
        for (int i = 0; i <= m + n + 1; i++)
            tata[i] = -1;
        for (int i = 0; i <= m + n + 1; i++)
            vizitat[i] = false;

    }

    g << cuplaje << '\n';

    for (int i = 1; i <= n; ++i) {
        for (int j = n + 1; j <= n + m; ++j) {
            if (rgraf[i][j] == 0 && rgraf[j][i] == 1)
                g << i << ' ' << j - n << '\n';
        }
    }
    f.close();
    f.close();
    return 0;
}