Pagini recente » Cod sursa (job #1638061) | Cod sursa (job #2614141) | Cod sursa (job #3209442) | Cod sursa (job #508385) | Cod sursa (job #2961638)
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#include <climits>
using namespace std;
// ALGORITM EDMONDS-KARP
ifstream fin("maxflow.in");
ofstream fout("maxflow.out");
int N, M;
// MATRICE DE ADIACENTA
vector<vector<int>> m_a;
// MATRICE DE CAPACITATE
vector<vector<int>> m_c;
// VECTOR DE TATI
vector<int> v_p;
bool BFS (int s, int d)
{
// VECTOR CARE VERIFICA DACA UN NOD A FOST SAU NU VIZITAT
vector<bool> vizitat(N + 1);
v_p[s] = -1;
vizitat[s] = true;
// COADA PENTRU PARCURGEREA GRAFULUI
queue<int> q;
q.push(s);
// CAT TIMP MAI AVEM NODURI IN COADA
while (!q.empty())
{
// PRIMUL NOD DIN COADA
int n_curent = q.front();
q.pop();
// PARCURGEM VECINII ACESTUIA
for (int n_vecin : m_a[n_curent])
{
// DACA NODUL NU A FOST VIZITAT & CAPACITATEA NU I-A FOST DEPASITA
if (!vizitat[n_vecin] && m_c[n_curent][n_vecin])
{
// ACTUALIZAM TATAL NODULUI VECIN CU NODUL CURENT
v_p[n_vecin] = n_curent;
// DACA AM AJUNS IN NODUL DESTINATIE, NE OPRIM
if(n_vecin == d) return true;
// DACA NU, VIZITAM NODUL VECIN & IL PUNEM IN COADA PENTRU CONTINUAREA PARCURGERII
vizitat[n_vecin] = true;
q.push(n_vecin);
}
}
}
// DACA NU AM AJUNS IN NODUL DESTINATIE, FLUXUL ESTE MAXIM
return false;
}
int MAXFLOW (int s, int d)
{
long long m_flow = 0;
// CAT TIMP AJUNGEM DIN NODUL SURSA IN NODUL DESTINATIE A.K.A
// CAT TIMP FLUXUL NU ESTE MAXIM
while (BFS(s, d))
{
// VALOAREA MINIMA A MUCHIILOR DE PE DRUMUL GASIT
int flow = INT_MAX;
for (int nod = d; nod != s; nod = v_p[nod])
flow = min(flow, m_c[v_p[nod]][nod]);
// ACTUALIZAM VALORILE MUCHIILOR DE PE DRUMUL GASIT SI A INVERSELOR LOR
for (int nod = d; nod != s; nod = v_p[nod])
{
// PENTRU MUCHIA ORIGINALA, II SCADEM CAPACITATEA
m_c[v_p[nod]][nod] -= flow;
// PENTRU MUCHIA INVERSATA, II MARIM CAPACITATEA
m_c[nod][v_p[nod]] += flow;
}
m_flow += flow;
}
return m_flow;
}
int main()
{
// CITIRE
fin>>N>>M;
// INITIALIZARE
int sursa = 1, destinatie = N;
m_a.resize(N + 1);
m_c.resize(N + 1, vector<int>(N + 1));
v_p.resize(N + 1);
for(int i = 0; i < M; i++)
{
int x, y, capacitate;
fin>>x>>y>>capacitate;
m_c[x][y] = capacitate;
m_a[x].push_back(y);
// MUCHIE PENTRU GRAFUL REZIDUAL
m_a[y].push_back(x);
}
// AFISARE
int m_flow = MAXFLOW(sursa, destinatie);
fout<<m_flow;
return 0;
}