Cod sursa(job #3196393)

Utilizator DrioanDragos Ioan Drioan Data 23 ianuarie 2024 19:48:56
Problema Flux maxim Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.35 kb
#include <iostream>
#include <fstream>
#include <utility>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
ifstream in("maxflow.in");
ofstream out("maxflow.out");

vector<int>ls[1005],ls1[1005];
int c[1005][1005], f[1005][1005];
int viz[101],tata[101];

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 main() {
	int n, s, t;
	in >> n;
	int m;
	in >> m;
	s = 1;
	t = n;
	while (m--) {
		int a, b, capacitate, flux=0;
		in >> a >> b >> capacitate ;
		ls[a].push_back(b);
		
		ls[b].push_back(a);
		c[a][b] += capacitate;
		
	}
	int total=0;
	for(;bfs(s,t);) {
		int flux_min = 999999999;
		for (int nod = t; nod != s; nod = tata[nod]) {
			flux_min = min(flux_min, c[tata[nod]][nod] - f[tata[nod]][nod]);
		}
		for (int nod = t; nod != s; nod = tata[nod]) {
			f[tata[nod]][nod] += flux_min;
			f[nod][tata[nod]] -= flux_min;
		}
		total += flux_min;
	}
	out<<total;
	
	
	
}