Cod sursa(job #2697711)

Utilizator FrostfireMagirescu Tudor Frostfire Data 19 ianuarie 2021 13:35:16
Problema Flux maxim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.28 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], ans;
queue <int> Q;

int addFlow()
{	for(int i=1; i<=n; i++)
		p[i] = 0;
	p[1] = 1;
	while(!Q.empty())
		Q.pop();
	Q.push(1);

	while(!Q.empty())
		{	int a = Q.front();
			Q.pop();
			for(int i=1; i<=n; i++)
				if(!p[i] && nod[a][i])
					{	p[i] = a;
						Q.push(i);
					}
		}
	return p[n];
}

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