Cod sursa(job #941345)

Utilizator howsiweiHow Si Wei howsiwei Data 18 aprilie 2013 16:18:09
Problema Flux maxim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.33 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
#define ech(It, A) for (typeof(A.begin()) It = A.begin(); It != A.end(); ++It)

vector< vector<int> > adjl; //have problem if exist multiple edges between two vertices
vector< vector<int> > adjm;
const int inf = 1<<30;

int main() {
	ifstream fin("maxflow.in");
	ofstream fout("maxflow.out");
	int n, m;
	fin >> n >> m;
	adjl.resize(n+1);
	adjm.assign( n+1, vector<int>(n+1) );

	for (int u, v, c, i = 0; i < m; ++i) {
		fin >> u >> v >> c;
		adjl[u].push_back(v);
		adjl[v].push_back(u);
		adjm[u][v] = c;
	}

	int flow = 0;

	while (true) {
		bool reach_sink = false;
		queue<int> Q;
		Q.push(1);
		vector<int> par(n+1);
		vector<bool> visited(n+1);
		visited[1] = true;

		while (!Q.empty()) {
			int u = Q.front();
			Q.pop();
			ech( it, adjl[u] ) {
				if (!visited[*it] && adjm[u][*it] > 0) {
					Q.push(*it);
					par[*it] = u;
					visited[*it] = true;
					if (*it == n) {
						reach_sink = true;
						break;
					}
				}
			}
			if (reach_sink) break;
		}
		if (!reach_sink) break;

		int cap = inf;
		for (int v = n; v != 1; v = par[v]) {
			cap = min( cap, adjm[par[v]][v] );
		}
		for (int v = n; v != 1; v = par[v]) {
			adjm[par[v]][v] -= cap;
			adjm[v][par[v]] += cap;
		}
		flow += cap;
	}

	fout << flow;
	return 0;
}