Cod sursa(job #2886488)

Utilizator Iulia14iulia slanina Iulia14 Data 7 aprilie 2022 20:06:26
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.55 kb
#include <bits/stdc++.h>

using namespace std;
const int N = 1005;
const int inf = 1e9;
int c[N][N];
int f[N][N];
int prv[N];
int viz[N];
vector <int> g[N];
int n, m;
int bfs()
{
    queue <int> q;
    for (int i = 1; i <= n; i++)
        viz[i] = 0;
    viz[1] = 1;
    q.push(1);
    while (!q.empty())
    {
        int x = q.front();
        q.pop();
        if (x == n)
            continue;
        for (auto y : g[x])
        {
            if (viz[y] || c[x][y] == f[x][y])
                continue;
            viz[y] = 1;
            prv[y] = x;
            q.push(y);
        }
    }
    return viz[n];
}
int main()
{
    freopen("maxflow.in", "r", stdin);
    freopen("maxflow.out", "w", stdout);
    cin >> n >> m;
    for (int i = 1; i <= m; i++)
    {
        int a, b, cp;
        cin >> a >> b >> cp;
        g[a].push_back(b);
        g[b].push_back(a);
        c[a][b] += cp;
    }
    int max_flow = 0;
    while (bfs())
    {
        for (int vec : g[n])
        {
            if (!viz[vec] || f[vec][n] == c[vec][n])
                continue;
            int fmin = inf;
            prv[n] = vec;
            for (int i = n; i != 1; i = prv[i])
                fmin = min(fmin, c[prv[i]][i] - f[prv[i]][i]);
            if (fmin == 0)
                continue;
            for (int i = n; i != 1; i = prv[i])
            {
                f[prv[i]][i] += fmin;
                f[i][prv[i]] -= fmin;
            }
            max_flow += fmin;
        }
    }
    cout << max_flow;
    return 0;
}