Pagini recente » Cod sursa (job #356959) | Cod sursa (job #1794517) | Cod sursa (job #1961730) | Cod sursa (job #232189) | Cod sursa (job #2964169)
#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;
}