Pagini recente » Cod sursa (job #3291247) | Cod sursa (job #5790) | Cod sursa (job #2736609) | Cod sursa (job #3293139) | Cod sursa (job #3260386)
#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
#include <fstream>
#include <climits>
using namespace std;
const int MAXN = 1005;
const int INF = INT_MAX;
// Matrice de adiacență pentru capacități
int capacity[MAXN][MAXN];
// Graful rezidual
int residual[MAXN][MAXN];
// Păstrează drumul pentru reconstrucția căii augmentante
int parent[MAXN];
// Implementarea BFS pentru a găsi o cale augmentantă
bool bfs(int source, int sink, int n) {
memset(parent, -1, sizeof(parent));
queue<int> q;
q.push(source);
parent[source] = source;
while (!q.empty()) {
int u = q.front();
q.pop();
for (int v = 1; v <= n; v++) {
// Dacă există capacitate reziduală și nodul nu a fost vizitat
if (parent[v] == -1 && residual[u][v] > 0) {
parent[v] = u;
if (v == sink) {
return true; // Am găsit o cale până la sink
}
q.push(v);
}
}
}
return false;
}
// Funcția principală pentru calculul fluxului maxim
int edmondsKarp(int source, int sink, int n) {
// Inițializăm graful rezidual cu capacitățile originale
memcpy(residual, capacity, sizeof(capacity));
int maxFlow = 0;
// Găsire căi augmentante
while (bfs(source, sink, n)) {
// Determinăm fluxul maxim ce poate fi adăugat pe această cale
int pathFlow = INF;
for (int v = sink; v != source; v = parent[v]) {
int u = parent[v];
pathFlow = min(pathFlow, residual[u][v]);
}
// Actualizăm graful rezidual
for (int v = sink; v != source; v = parent[v]) {
int u = parent[v];
residual[u][v] -= pathFlow;
residual[v][u] += pathFlow;
}
// Adăugăm fluxul pe calea augmentantă
maxFlow += pathFlow;
}
return maxFlow;
}
int main() {
// File I/O streams
ifstream inFile("maxflow.in");
ofstream outFile("maxflow.out");
if (!inFile || !outFile) {
cerr << "Error opening file." << endl;
return 1;
}
int n, m;
inFile >> n >> m;
memset(capacity, 0, sizeof(capacity));
for (int i = 0; i < m; i++) {
int x, y, z;
inFile >> x >> y >> z;
capacity[x][y] = z; // Adăugăm capacitatea pe muchia x -> y
}
// Calculăm fluxul maxim de la sursă (1) la destinație (n)
int source = 1;
int sink = n;
int result = edmondsKarp(source, sink, n);
outFile << result << endl;
inFile.close();
outFile.close();
return 0;
}