Cod sursa(job #2697702)

Utilizator Iulia25Hosu Iulia Iulia25 Data 19 ianuarie 2021 13:10:08
Problema Flux maxim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <fstream>
#include <cstring>
#include <vector>
#include <queue>

using namespace std;

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

queue <int> q;
vector <int> v[1005];

int c[1005][1005];

int anterior[1005], cost[1005];
int ans, n, m;

void reinit()  {
  memset(anterior, 0, sizeof(anterior));
  memset(cost, 0, sizeof(cost));
  cost[1] = 2e9;
  while (!q.empty())
    q.pop();
}

void bfs();

void reconstituire()  {
  int nod = n;
  while (nod != 1)  {
    c[anterior[nod]][nod] -= cost[n];
    c[nod][anterior[nod]] += cost[n];
    nod = anterior[nod];
  }
  ans += cost[n];
  bfs();
}

void bfs()  {
  reinit();
  q.push(1);
  while (!q.empty())  {
    int nod = q.front();
    q.pop();
    for (auto it : v[nod])  {
      if (c[nod][it] == 0 || cost[it])
        continue;
      cost[it] = min(cost[nod], c[nod][it]);
      anterior[it] = nod;
      q.push(it);
      if (it == n)  {
        if (cost[n] > 0)
          reconstituire();
        return;
      }
      if (c[it][n] > 0)  {
        cost[n] = min(cost[it], c[it][n]);
        anterior[n] = it;
        if (cost[n] > 0)
          reconstituire();
      }
    }
  }
}

int main()  {
  cin >> n >> m;
  int x, y, z;
  for (int i = 1; i <= m; ++i)  {
    cin >> x >> y >> z;
    c[x][y] = z;
    v[x].push_back(y);
    v[y].push_back(x);
  }
  bfs();
  cout << ans;
	return 0;
}