Cod sursa(job #1396926)

Utilizator irimiecIrimie Catalin irimiec Data 23 martie 2015 09:47:58
Problema Flux maxim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.88 kb
#include <bits/stdc++.h>

using namespace std;

#define     mp              make_pair
#define     fs              first
#define     sc              second
#define     pob             pop_back
#define     pub             push_back
#define     eps             1E-7
#define     sz(a)           a.size()
#define     count_one       __builtin_popcount;
#define     count_onell     __builtin_popcountll;
#define     fastIO          ios_base::sync_with_stdio(false)
#define     PI              (acos(-1.0))
#define     linf            (1LL<<62)//>4e18
#define     inf             (0x7f7f7f7f)//>2e9

#define MAXN 1010

#ifndef ONLINE_JUDGE
ifstream fin("maxflow.in");
ofstream fout("maxflow.out");
#endif

int n, m;
int minflow;
int fn[MAXN];
int f[MAXN][MAXN];
int cap[MAXN][MAXN];
vector<int> vec[MAXN];

bool BFS() {
    int nod;
    bitset<MAXN> viz;
    queue<int> Q;
    Q.push(1);

    while(!Q.empty()) {
        nod = Q.front(); Q.pop();

        for(auto it : vec[nod]) {
            if(cap[nod][it] != f[nod][it] && !viz[it]) {
                viz[it] = 1;
                Q.push(it);
                fn[it] = nod;

                if(it == n)
                    return true;
            }
        }
    }
    return false;
}

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

    int a, b, c;
    for(int i = 0; i < m; ++i) {
        fin >> a >> b >> c;
        vec[a].pub(b);
        vec[b].pub(a);
        cap[a][b] += c;
    }

    int flow = 0;
    while(BFS()) {
        minflow = inf;

        for(int nod = n; nod != 1; nod = fn[nod]) {
            minflow = min(minflow, cap[fn[nod]][nod] - f[fn[nod]][nod]);
        }

        for(int nod = n; nod != 1; nod = fn[nod]) {
            f[fn[nod]][nod] += minflow;
            f[nod][fn[nod]] -= minflow;
        }

        flow += minflow;
    }

    fout << flow << "\n";

    return 0;
}