Cod sursa(job #3196406)

Utilizator DrioanDragos Ioan Drioan Data 23 ianuarie 2024 19:59:06
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.49 kb
#include <iostream>
#include <fstream>
#include <utility>
#include <vector>
#include <queue>
#include <cstring>
#define inf 0x3f3f3f3f
using namespace std;

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


#define nmax 1005
using namespace std;
vector<int>ls[nmax];
int n, m, c[nmax][nmax], f[nmax][nmax], maxflow, tata[nmax];
bool viz[nmax];

bool bfs(int s,int t)
{
	bool ok = false;
	queue<int> coada;
	memset(viz, 0, sizeof(viz));
	viz[s] = true;
	coada.push(s);
	while (!coada.empty())
	{
		int sursa = coada.front();
		coada.pop();
		for (auto vecin : ls[sursa])
		{
			if (!viz[vecin] && c[sursa][vecin] != f[sursa][vecin])
			{
				viz[vecin] = true;
				coada.push(vecin);
				tata[vecin] = sursa;
				if (vecin == t)
				{
					ok = true;
					break;
				}
			}
		}
	}
	return ok;
}


int edmondsKarp(int s, int t) {
	int total = 0;
	for (; bfs(s, t);)
	{
		int mincaprez = inf;
		for (int nod = n; nod != 1; nod = tata[nod])
		{
			mincaprez = min(mincaprez, c[tata[nod]][nod] - f[tata[nod]][nod]);
		}
		for (int nod = n; nod != 1; nod = tata[nod])
		{
			f[tata[nod]][nod] += mincaprez;
			f[nod][tata[nod]] -= mincaprez;
		}
		total += mincaprez;
	}
	return total;
}


int main() {
	in >> n >> m;
    for (int i = 1; i <= m; i++)
    {
        int a, b, z;
        in >> a >> b >> z;
        ls[a].push_back(b);
        ls[b].push_back(a);
        c[a][b] += z;
    }
	int s = 1, t = n;
	out << edmondsKarp(1, n);
	
	
	
}