Cod sursa(job #2697659)

Utilizator FrostfireMagirescu Tudor Frostfire Data 19 ianuarie 2021 11:32:20
Problema Flux maxim Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <iostream>
#include <fstream>
#include <queue>
#define f first
#define s second
#define NMAX 1000
#define inf 2000000000

using namespace std;

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

int n, m, nod[NMAX+10][NMAX+10], p[NMAX+10], viz[NMAX+10], ans;
queue <pair <int, int> > Q;

int addFlow()
{	for(int i=1; i<=n; i++)
		p[i] = 0, viz[i] = 0;
	while(!Q.empty())
		Q.pop();
	Q.push({1, inf});
	viz[1] = 1;
	while(!Q.empty())
		{	pair <int, int> a = Q.front();
			Q.pop();
			for(int i=1; i<=n; i++)
				if(!viz[i] && nod[a.f][i])
					{	p[i] = a.f;
						viz[i] = 1;
						int newFlow = min(a.s, nod[a.f][i]);
						if(i == n)
							return newFlow;
						Q.push({i, newFlow});
					}
		}
	return 0;
}

int main()
{	
	fin >> n >> m;
	for(int i=1; i<=m; i++)
		{	int nod1, nod2, c;
			fin >> nod1 >> nod2 >> c;
			nod[nod1][nod2] = c;
			nod[nod2][nod1] = 0;
		}
	int ok = 1;
	while(ok)
		{	ok = 0;
			int newFlow = addFlow();
			if(!newFlow) continue;
			ok = 1;
			ans += newFlow;
			int curr = n;
			while(curr != 1)
				{	int pr = p[curr];
					nod[pr][curr] -= newFlow;
					nod[curr][pr] += newFlow;
					curr = pr;
				}
		}
	fout << ans << '\n';
	return 0;
}