Cod sursa(job #1350931)

Utilizator ELHoriaHoria Cretescu ELHoria Data 21 februarie 2015 00:21:21
Problema Flux maxim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.89 kb
#include <algorithm>
#include <vector>
#include <climits>
#include <fstream>
using namespace std;

class MaxFlow {

		static const int maxnodes = 1000;
		int nodes, src, dest;
		int dist[maxnodes], q[maxnodes], work[maxnodes];

		struct Edge {
			int to, rev;
			int f, cap;
		};

		vector<Edge> g[maxnodes];


		bool dinic_bfs() {
			fill(dist, dist + nodes, -1);
			dist[src] = 0;
			int qt = 0;
			q[qt++] = src;
			for (int qh = 0; qh < qt; qh++) {
				int u = q[qh];
				for (int j = 0; j < (int)g[u].size(); j++) {
					Edge &e = g[u][j];
					int v = e.to;
					if (dist[v] < 0 && e.f < e.cap) {
						dist[v] = dist[u] + 1;
						q[qt++] = v;
					}
				}
			}
			return dist[dest] >= 0;
		}


		int dinic_dfs(int u, int f) {
			if (u == dest)
				return f;
			for (int &i = work[u]; i < (int)g[u].size(); i++) {
				Edge &e = g[u][i];
				if (e.cap <= e.f) continue;
				int v = e.to;
				if (dist[v] == dist[u] + 1) {
					int df = dinic_dfs(v, min(f, e.cap - e.f));
					if (df > 0) {
						e.f += df;
						g[v][e.rev].f -= df;
						return df;
					}
				}
			}
			return 0;
		}

	public:

		MaxFlow(int n,int src_,int dest_) {
			nodes = n;
			src = src_;
			dest = dest_;
		}

		void addEdge(int s, int t, int cap){
			Edge a = { t, g[t].size(), 0, cap };
			Edge b = { s, g[s].size(), 0, cap };
			g[s].push_back(a);
			g[t].push_back(b);
		}


		int maxFlow() {
	
			int result = 0;
			while (dinic_bfs()) {
				fill(work, work + nodes, 0);
				while (int delta = dinic_dfs(src, INT_MAX))
					result += delta;
			}
			return result;
		}

};
int main()
{
	ifstream cin("maxflow.in");
	ofstream cout("maxflow.out");
	int n, m;
	cin >> n >> m;
	MaxFlow solver(n, 0, n - 1);
	for (int i = 0; i < m; i++) {
		int a, b, c;
		cin >> a >> b >> c;
		solver.addEdge(--a, --b, c);
	}

	cout << solver.maxFlow();
	return 0;
}