Cod sursa(job #2950570)

Utilizator RobertuRobert Udrea Robertu Data 4 decembrie 2022 10:17:19
Problema Flux maxim Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.61 kb
#include <bits/stdc++.h>
#define dim 1010
#define inf INT_MAX
using namespace std;
ifstream fin("maxflow.in");
ofstream fout("maxflow.out");

long long maxFlow = 0;
int tata[dim], a[dim][dim], n, m, x, y, z, nod, flow;
bool viz[dim];
queue< pair<int, int> > coada;  //nodul si capacitatea minima pana la el

bool bfs() {
    while(!coada.empty()) {
        coada.pop();
    }
    memset(tata, 0, sizeof(tata));
    memset(viz, false, sizeof(viz));

    coada.push(make_pair(1, inf));
    viz[1] = true;
    while(!coada.empty()) {
        pair<int, int> front = coada.front();
        coada.pop();

        nod = front.first;
        flow = front.second;

        if(nod == n) {
            return true;
        }

        for(int i = 1; i <= n; i++) {
            if( !viz[i] &&  a[nod][i] > 0) {
                coada.push(make_pair(i, min(flow, a[nod][i])));
                viz[i] = true;
                tata[i] = nod;
            }
        }
    }
    return false;
}

int main() {
    fin >> n >> m;

    for(int i = 0; i < m; i++) {
        fin >> x >> y >> z;
        a[x][y] = z;
    }

    while(bfs()) {
        for(int i = 1; i < n; i++) {
            if( viz[i] && a[i][n] > 0 ) {
                flow = inf;
                for(nod = n; tata[nod] > 0; nod = tata[nod]) {
                    flow = min(flow, a[ tata[nod] ][nod]);
                }

                for(nod = n; tata[nod] > 0; nod = tata[nod]) {
                    a[ tata[nod] ][nod] -= flow;
                    a[nod][ tata[nod] ] += flow;
                }

                maxFlow += flow;
            }
        }
    }

    fout << maxFlow << '\n';

    return 0;
}