Cod sursa(job #1312174)

Utilizator thesvcoolmanLucian Bicsi thesvcoolman Data 8 ianuarie 2015 23:07:03
Problema Flux maxim Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.56 kb
#include<fstream>
#include<vector>
#include<deque>
#define MAXN 1001

using namespace std;

ifstream fin("maxflow.in");
ofstream fout("maxflow.out");

int G[MAXN][MAXN];
int n, m, source, destination;

vector<int> PATH;
vector<bool> VIZ;

void afis(vector<int> &v) {
    for(int i=0; i<v.size(); i++) {
        fout<<v[i]<<" ";
    }fout<<endl;
}

void read() {
    fin>>n>>m;
    int a, b;
    VIZ.resize(n+1, false);

    source = 1;
    destination = n;

    for(int i=1; i<=m; i++) {
        fin>>a>>b;
        fin>>G[a][b];
    }

}

bool done;
void getPath (int s, int d) {
    VIZ[s] = true;
    PATH.push_back(s);

    if(s == d) {
        done = true;
        return;
    }

    for(int i=1; i<=n; i++) {
        if(G[s][i] and !VIZ[i]) {
            getPath(i, d);
            if(done) return;
        }
    }
    PATH.pop_back();
}

void resGraph() {
    int MIN = G[PATH[0]][PATH[1]];
    for(int i=2; i<PATH.size(); i++) {
        MIN = min(MIN, G[PATH[i-1]][PATH[i]]);
    }
    for(int i=1; i<PATH.size(); i++) {
        G[PATH[i-1]][PATH[i]] -= MIN;
        G[PATH[i]][PATH[i-1]] += MIN;
    }
}

int main() {
    read();
    getPath(source, destination);
    while(done) {
        resGraph();
        done = false;
        for(int i=1; i<=n; i++) {
            VIZ[i] = false;
        }
        //afis(PATH);
        PATH.clear();
        getPath(source, destination);
    }

    int flux = 0;
    for(int i=1; i<=n; i++) {
        flux +=G[i][source];
    }
    fout<<flux;

    return 0;
}