Cod sursa(job #2954900)

Utilizator IRadu1529Radu Ionescu IRadu1529 Data 15 decembrie 2022 19:17:04
Problema Cuplaj maxim in graf bipartit Scor 32
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.84 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <fstream>
#include <set>
using namespace std;


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

#define N 20001

int n, m, e, maxFlow;
int s, t;
int capacity[N][N];
vector<vector<int>> l(N, vector<int>());
vector<int> parent(N);
set<pair<int, int>> edges;
                                                                    //O(N * M2)
int bfs(int s, int t) {
    parent[s] = -2; // orice != -1 si [0,n+m+1]
    queue<pair<int, int>> q;
    q.push({ s, 1e9 });

    while (!q.empty()) {
        int cur = q.front().first;
        int maxFlow = q.front().second;
        q.pop();

        for (int next : l[cur]) {
            if (parent[next] == -1 && capacity[cur][next]) {
                parent[next] = cur;
                int bottleneck = min(maxFlow, capacity[cur][next]);
                if (next == t)
                    return bottleneck;
                q.push({ next, bottleneck });
            }
        }
    }

    return 0;
}

void read() {
    fin >> n >> m >> e;

    while (e--) {
        int a, b;
        fin >> a >> b; // muchia a -> n+b
        l[a].push_back(n + b);
        l[n + b].push_back(a);
        capacity[a][n + b] = 1; // muchia de inaintare 
    }
}

int main() {

    read();

    s = 0; t = n+m+1;

    for (int i = 1; i <= n; i++) { // concectam sursa la nodurile din A
        l[s].push_back(i);
        l[i].push_back(s);
        capacity[s][i] = 1; 
    }

    for (int i = n + 1; i <= n+m; i++) { // concectam nodurile din B la destinatie
        l[i].push_back(t);
        l[t].push_back(i);
        capacity[i][t] = 1;
    }

    for (int i = 0; i <= n+ m + 1; i++) // resetam parintii
        parent[i] = -1;

    int bottleneck;

    set<pair<int, int>> edges;

    while (bottleneck = bfs(s, t)) { // cat mai sunt path uri care mai accepta flux
        maxFlow += bottleneck;
        int cur = t;
        while (cur != s) { // mergem inapoi pe path ul gasit 
            int prev = parent[cur];
            //cout << prev << " " << cur << "\n";                
            capacity[prev][cur] -= bottleneck; // muchia de inaintare pierde din spatiu exact bottleneck
            capacity[cur][prev] += bottleneck; // muchia de retur poate face undo cu exact bottleneck mai mult

            if(capacity[prev][cur] == 0 && cur > s && cur < t && prev > s && prev < t && cur > prev) // doar muchii de inaintare dintre A si B
                edges.insert({ prev, cur - n });

            cur = prev;

        }
        //cout << "\n\n";
        for (int i = 0; i <= n + m + 1; i++) // resetam parintii
            parent[i] = -1;
    }

    fout << maxFlow << "\n";
 

    for (auto i : edges)
        fout << i.first << " " << i.second << "\n";
    
    return 0;
}