Cod sursa(job #2066031)

Utilizator silviu982001Borsan Silviu silviu982001 Data 14 noiembrie 2017 17:44:46
Problema Flux maxim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.79 kb
#include <fstream>
#include <unordered_set>
#include <iostream>
#include <limits.h>
#include <string.h>
#include <queue>
#include <map>
#include <vector>

#define MAX 999999
using namespace std;

int c[1000][1000];
map<int, vector<int>> nodes;


int main()
{
	ifstream in;
	in.open("maxflow.in", ios::in);
	int n, m, u, v, source, dest;
	in >> n >> m;
	source = 0;
	dest = n - 1;
	for (int i = 0; i < m; i++)
	{
		in >> u >> v;
		u--;
		v--;
		nodes[u].push_back(v);
		nodes[v].push_back(u);
		in >> c[u][v];
	}
	in.close();
	unordered_set<int> visited;
	int parent[1000];
	long max_flow = 0;

	for (;;)
	{
		memset(parent, 0, sizeof(parent));
		parent[source] = -1;
		visited.clear();
		visited.insert(source);
		queue<int> q;
		q.push(source);

		while (visited.find(dest) == visited.end() && !q.empty())
		{
			int nod = q.front();
			q.pop();
			for (vector<int>::const_iterator it = nodes[nod].begin(); it != nodes[nod].end(); it++)
			{
				if (visited.find(*it) == visited.end() && c[nod][*it])
				{
					parent[*it] = nod;
					visited.insert(*it);
					q.push(*it);
				}
			}
		}

		if (visited.find(dest) == visited.end())
			break;

		int min_f = MAX;
		for (v = dest; v != source; v = parent[v])
		{
			u = parent[v];
			if (c[u][v] < min_f)
				min_f = c[u][v];
		}

		max_flow = max_flow + min_f;

		for (v = dest; v != source; v = parent[v])
		{
			u = parent[v];
			c[u][v] = c[u][v] - min_f;
			if (c[u][v] <= 0)
			{
				nodes[u].erase(std::find(nodes[u].begin(), nodes[u].end(), v));
			}
			c[v][u] = c[v][u] + min_f;
			if (std::find(nodes[v].begin(), nodes[v].end(), u) == nodes[v].end())
			{
				nodes[v].push_back(u);
			}
		}
	}

	ofstream out("maxflow.out", ios::out);
	out << max_flow;
	out.close();
	return 0;
}