Cod sursa(job #1678359)

Utilizator serbanSlincu Serban serban Data 7 aprilie 2016 11:24:58
Problema Flux maxim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include <iostream>
#include <fstream>
#include <queue>
#define oo 2147483647
using namespace std;

int n;
int c[1005][1005];

bool vis[1005];
int tata[1005];

int minCap(int in) {
    int i = in;
    int mn = c[i][n];
    for(; tata[i]; i = tata[i]) {
        mn = min(mn, c[tata[i]][i]);
    }
    c[in][n] -= mn;
    c[n][in] += mn;
    for(; tata[in]; in = tata[in]) {
        c[tata[in]][in] -= mn;
        c[in][tata[in]] += mn;
    }
    return mn;
}

int onPath() {
    queue<int> q;
    while(!q.empty()) q.pop();
    q.push(1);
    vis[1] = true;
    while(!q.empty()) {
        int cur = q.front();
        q.pop();
        if(c[cur][n] > 0) {
            return minCap(cur);
        }
        for(int i = 2; i < n; i ++) {
            if(c[cur][i] > 0) {
                vis[i] = true;
                tata[i] = cur;
                q.push(i);
            }
        }
    }
    return oo;
}

int flow;
void maxFlow() {
    int capacity = 0;
    while(true) {
        capacity = onPath();
        if(capacity == oo) break;
        flow += capacity;
    }
}

int main()
{
    ifstream f("maxflow.in");
    ofstream g("maxflow.out");
    int m, x, y;
    f >> n >> m;
    for(int i = 1; i <= m; i ++) {
        f >> x >> y; f >> c[x][y];
        if(!c[y][x]) c[y][x] = c[x][y];
    }

    maxFlow();
    g << flow << "\n";
    return 0;
}