Cod sursa(job #592071)

Utilizator darrenRares Buhai darren Data 26 mai 2011 18:08:31
Problema Flux maxim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.67 kb
#include <cstring>
#include <fstream>
#include <algorithm>
#include <queue>

using namespace std;

const int INF = 1 << 30;

int maxflow, n, m;
int c[1001][1001], f[1001][1001], cr[1001], t[1001];
bool s[1001];
queue<int> q;
vector<int> v;

inline int abs(int x)
{
	return x < 0 ? -x : x;
}
bool bfs();
void dim();

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

	fin >> n >> m;
	for (int i = 1, nod1, nod2, cap; i <= m; ++i)
	{
		fin >> nod1 >> nod2 >> cap;
		c[nod1][nod2] = cap;
	}

	while (bfs())
		dim();

	fout << maxflow;
}

bool bfs()
{
    v.clear();
	memset(s, 0, sizeof(s));
	memset(cr, 0, sizeof(cr));
	memset(t, 0, sizeof(t));
	cr[1] = INF;

	while (!q.empty()) q.pop();
	q.push(1);
	while (!q.empty())
	{
		int i = q.front(); q.pop();

		if (f[i][n] < c[i][n])
		{
		    v.push_back(i);
            continue;
		}
		for (int j = 1; j <= n; ++j)
			if (!s[j])
			{
				if (f[i][j] < c[i][j])
				{
					cr[j] = min(c[i][j] - f[i][j], cr[i]);
					t[j] = i, s[j] = true;
					q.push(j);
				}
				if (f[j][i] > 0)
				{
					cr[j] = min(f[j][i], cr[i]);
					t[j] = -i, s[j] = true;
					q.push(j);
				}
			}
	}

	if (v.size()) return true;
	return false;
}

void dim()
{
    while (!v.empty())
    {
        int i = v.back();
        int crnow = min(cr[i], c[i][n] - f[i][n]);
        f[i][n] += crnow;

        int j = i;
        while (j != 1)
        {
            i = abs(t[j]);
            if (t[j] > 0) f[i][j] += crnow;
            else          f[j][i] -= crnow;
            j = abs(t[j]);
        }
        maxflow += crnow;

        v.pop_back();
    }
}