Cod sursa(job #3260386)

Utilizator Zh19Draghici Alexandru Andrei Zh19 Data 2 decembrie 2024 00:46:55
Problema Flux maxim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.69 kb
#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;
}