Cod sursa(job #2763875)

Utilizator DragosC1Dragos DragosC1 Data 17 iulie 2021 14:54:04
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.8 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;

int n, m;
vector<int> a[1001];
int C[1001][1001];
int F[1001][1001];
int t[1001];
int maxflow;
bool viz[1001];

const int Inf = 1e9;

void read() {
    int i, x, y, cost;
    ifstream f("maxflow.in");
    f >> n >> m;
    for (i = 1; i <= m; i++) {
        f >> x >> y >> cost;
        a[x].emplace_back(y);
        a[y].emplace_back(x);
        C[x][y] += cost;
    }
    f.close();
}

queue<int> Q;

bool bfs() {
    int i, x, fiu;
    for (i = 1; i <= n; i++)
        viz[i] = 0;
    viz[1] = 1;
    Q.push(1);
    while (!Q.empty()) {
        x = Q.front();
        Q.pop();
        if (x == n)
            continue;
        for (i = 0; i < a[x].size(); i++) {
            fiu = a[x][i];
            if (viz[fiu] || C[x][fiu] == F[x][fiu])
                continue;
            viz[fiu] = 1;
            t[fiu] = x;
            Q.push(fiu);
        }
    }
    return viz[n];
}

void solve() {
    int flow, i, fiu, cur;
    while (bfs()) 
        for (i = 0; i < a[n].size(); i++) {
            fiu = a[n][i];
            if (!viz[fiu] || C[fiu][n] == F[fiu][n])  
                continue;
            t[n] = fiu;
            flow = Inf;
            cur = n;
            while (cur != 1) {
                flow = min(flow, C[t[cur]][cur] - F[t[cur]][cur]);
                cur = t[cur];
            }
            cur = n;
            while (cur != 1) {
                F[t[cur]][cur] += flow;
                F[cur][t[cur]] -= flow;
                cur = t[cur];
            }
            maxflow += flow;
        }
}

void output() {
    ofstream g("maxflow.out");
    g << maxflow;
    g.close();
}

int main() {
    read();
    solve();
    output();
    return 0;
}