Cod sursa(job #2727668)

Utilizator vlad082002Ciocoiu Vlad vlad082002 Data 22 martie 2021 12:13:58
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.35 kb
#include <bits/stdc++.h>
using namespace std;

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

int n, m, c[1024][1024], f[1024][1024], t[1024], v[1024];
vector<int> g[1024];

bool bfs() {
    for(int i = 1; i <= n; i++) v[i] = false;

    queue<int> q;
    q.push(1);
    v[1] = true;
    while(!q.empty()) {
        int cur = q.front();
        q.pop();

        for(auto next: g[cur]) {
            if(!v[next] && c[cur][next] > f[cur][next]) {
                t[next] = cur;
                v[next] = true;
                if(next == n) return true;
                q.push(next);
            }
        }
    }
    return false;
}

int main() {
    fin >> n >> m;
    while(m--) {
        int x, y, z;
        fin >> x >> y >> z;
        g[x].push_back(y);
        g[y].push_back(x);
        c[x][y] = z;
    }

    int flow = 0;
    for( ; bfs(); ) {
        for(auto tt: g[n]) {
            int fmin = 2e9;
            if(!v[tt] || c[tt][n] <= f[tt][n]) continue;
            t[n] = tt;
            for(int i = n; i != 1; i = t[i]) {
                fmin = min(fmin, c[t[i]][i]-f[t[i]][i]);
            }
            for(int i = n; i != 1; i = t[i]) {
                f[t[i]][i] += fmin;
                f[i][t[i]] -= fmin;
            }
            flow += fmin;
        }
    }
    fout << flow << '\n';
}