Cod sursa(job #2695767)

Utilizator raresmateiuMateiu Rares-Ioan raresmateiu Data 14 ianuarie 2021 14:39:51
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.87 kb
#include <bits/stdc++.h>
using namespace std;

ifstream f ("maxflow.in");
ofstream g ("maxflow.out");

#define N 1001
#define INF 110001

int adj[N][N];
int n, m;
vector<int> path(N, 0);
vector<int> v[N];

void bfs()
{
    queue<int> q;

    for(int i = 1; i <= n; ++i)
        path[i] = 0;

    q.push(1);
    path[1] = -1;


    while(!q.empty() && !path[n])
    {
        int curr = q.front();
        q.pop();
        for(auto j: v[curr])
        {

            if(adj[curr][j] && !path[j])
            {
                path[j] = curr;
                q.push(j);
            }
        }

    }
}

int ford_fulkerson()
{
    int ans = 0;
    while(true)
    {
        bfs();

        if(path[n] == 0)
            return ans;
        else
        {
                for(auto j: v[n])
                {
                    int curr = n;
                    path[n] = j;
                    if(path[j] > 0)
                    {
                        int flow = INF;

                        while(path[curr] != -1)
                        {
                            flow = min(flow, adj[path[curr]][curr]);
                            curr = path[curr];
                        }
                        curr = n;

                        while(path[curr] != -1)
                        {
                            adj[path[curr]][curr] -= flow;
                            adj[curr][path[curr]] += flow;
                            curr = path[curr];
                        }
                        ans += flow;
                    }
                }
        }
    }
}
int main()
{
    f >> n >> m;
    for(int i = 0; i < m ; ++i)
    {
        int a, b, c;
        f >> a >> b >> c;
        adj[a][b] = c;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    g << ford_fulkerson();
    return 0;
}