Cod sursa(job #2964169)

Utilizator Ruxi_GontescuGontescu Maria Ruxandra Ruxi_Gontescu Data 12 ianuarie 2023 15:33:11
Problema Cuplaj maxim in graf bipartit Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.71 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

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

struct edge{
    int node1, node2, cap, pos;
};
vector<vector<int> > adjList;
vector<edge> edges;
vector<bool> visited;
vector<int> parent;
edge muchie;
int n, m, e, nrN, source, dest;
void addEdge(int x, int y){
    int dim = edges.size();//cate muchii au fost deja adaugate
    // se adauga muchia de capacitate 1
    adjList[y].push_back(dim+1);
    muchie.node1 = x;
    muchie.node2 = y;
    muchie.cap = 1;
    muchie.pos = dim + 1;
    edges.push_back(muchie);

    // muchia inversa de capacitate 0
    adjList[x].push_back(dim);
    muchie.node1 = y;
    muchie.node2 = x;
    muchie.cap = 0;
    muchie.pos = dim;
    edges.push_back(muchie);
}
// true =  a gasit drum de ameliorare
bool bfs (){
    // arborele bfs se creaza prin vectorul de parinti
    parent.clear();
    parent.resize(nrN);
    visited.clear();
    visited.resize(nrN, false);

    visited[source] = true;
    queue<int> q;
    q.push(source);
    while (!q.empty()){
        // nodul parinte
        int pNode = q.front();
        q.pop();
        if (pNode != dest){
            // pozitia din lista de muchii a nodului parinte
            for (int index : adjList[pNode]){
                edge cEdge = edges[index];
                if (!visited[cEdge.node2] && cEdge.cap != 0){
                    parent[cEdge.node2] = index;
                    // adaug urmatorul nod ce urmeaza sa fie vizitat
                    q.push(cEdge.node2);
                    visited[cEdge.node2] = true;
                }
            }
        }
    }
    return visited[dest];
}

int FordFulkerson (int source, int dest){
    int maxFlow = 0;
    // atata timp cat mai exista drumuri de ameliorare, inseamnca ca putem adauga flux
    while (bfs()){
        for (int index : adjList[dest]){
            if (visited[edges[index].node2] && edges[edges[index].pos].cap != 0){
                int minCap = 2e9;
                edge cEdge = edges[index];
                parent[dest] = cEdge.pos;
                int cNode = dest;
                while (cNode != source){
                    minCap = min(minCap, edges[parent[cNode]].cap);
                    cNode = edges[parent[cNode]].node1;
                }
                cNode = dest;
                // se parcurge drumul de ameliorare pt a adauga fluxul nou
                while (cNode != source){
                    edges[edges[parent[cNode]].pos].cap += minCap;
                    edges[parent[cNode]].cap -= minCap;
                    cNode = edges[parent[cNode]].node1;
                }
                maxFlow += minCap;
            }
        }
    }
    return maxFlow;
}
int main() {
    int x, y, i;
    fin >> n >> m >> e;
    // Numarul de noduri.
    nrN = n + m + 2;
    source = 0;
    dest = nrN - 1;
    adjList.resize(nrN);
    for (int i = 0; i < e; ++i) {
        fin >> x >> y;
        addEdge(x, y + n);
        // y + n = nodurile din multimea din dreapta sunt numerotate : curent + cateva au fost in stanga
    }
    for (edge edg : edges)
        cout << edg.node1 << "," << edg.node2 << " " << edg.cap << " " << edg.pos << " ; ";
    cout << endl;
    // se adauga  muchiile de la sursa la toate nodurile din stanga
    for (i=1; i<=n; i++)
        addEdge(source, i);
    // muchiile de la toate nodurile din dreapa la destinatie
    for (i=1; i<=m; i++)
        addEdge(i+n, dest);
    // afisare cuplaj maxim
    fout << FordFulkerson(source, dest) << endl;
    // cuplajele
    for (edge edg : edges)
        if (edg.node1 < edg.node2 && edg.node1 != 0 && edg.node2 != dest && edg.cap == 0)
            fout << edg.node1 << " " << edg.node2 - n << endl;
    fout.close();
    return 0;
}