Cod sursa(job #2391480)

Utilizator Constantin.Dragancea Constantin Constantin. Data 28 martie 2019 21:31:40
Problema Flux maxim Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.56 kb
#include <bits/stdc++.h>
using namespace std;

/*
int n, m, c[N][N], p[N];
bool viz[N];
queue <int> Q;
vector <int> v[N];

bool bfs(int s, int t){
    for (int i=1; i<=n; i++) viz[i] = 0;
    viz[s] = 1; Q.push(s);
    while (!Q.empty()){
        int nod = Q.front();
        Q.pop();
        if (nod == n) continue;
        for (auto it: v[nod]){
            if (viz[it] || !c[nod][it]) continue;
            viz[it] = 1; Q.push(it); p[it] = nod;
        }
    }
    return viz[t];
}

int EdmondsKarp(int s, int t){
    int maxflow = 0;
    while (bfs(s, t)){
        for (auto it: v[n]){
            if (!viz[it] || !c[it][n]) continue;
            int flow = 1e9;
            p[n] = it;
            for (int nod = t; nod != s; nod = p[nod])
                flow = min(flow, c[p[nod]][nod]);
            for (int nod = t; nod != s; nod = p[nod]){
                c[p[nod]][nod] -= flow;
                c[nod][p[nod]] += flow;
            }
            maxflow += flow;
        }
    }
    return maxflow;
}
*/

#define N 1005
int n, m, c[N][N], fv[2*N];
struct node{
    int eflow, h;
} ver[N];
bool inq[N];
queue <int> Q;

void preFlow(int s){
    for (int i=1; i<=n; i++) ver[i] = {0, 0};
    ver[s].h = n; fv[n]++;
    for (int i=1; i<=n; i++){
        if (i != s && c[s][i]) Q.push(i), ver[i].eflow += c[s][i], c[i][s] = c[s][i], c[s][i] = 0, inq[i] = 1;
    }
}

void push(int q){
    for (int i=1; i<=n; i++){
        if (i == q || !c[q][i] || ver[i].h >= ver[q].h) continue;
        int flow = min(c[q][i], ver[q].eflow);
        ver[q].eflow -= flow;
        ver[i].eflow += flow;
        c[q][i] -= flow;
        c[i][q] += flow;
        if (!inq[i]) Q.push(i), inq[i] = 1;
    }
}

void normalize_h(int qh){
    if (!fv[qh]){
        for (int i=1; i<=n; i++)
            if (ver[i].h > qh) fv[ver[i].h]--, ver[i].h = qh, fv[qh]++;
    }
}

void relabel(int q){
    fv[ver[q].h]--;
    fv[++ver[q].h]++;
    normalize_h(ver[q].h - 1);
}

int push_relabel(int s, int t){
    preFlow(s);
    while (!Q.empty()){
        int u = Q.front();
        Q.pop(); inq[u] = 0;
        if (!ver[u].eflow || u == s || u == t) continue;
        while (ver[u].eflow){
            push(u);
            if (ver[u].eflow) relabel(u);
        }
    }
    return ver[t].eflow;
}

int main(){
    ifstream cin ("maxflow.in");
    ofstream cout ("maxflow.out");
    cin >> n >> m;
    for (int i=1, x, y, z; i<=m; i++){
        cin >> x >> y >> z;
        c[x][y] += z;
    }
    cout << push_relabel(1, n);
    return 0;
}