Cod sursa(job #2818507)

Utilizator victorzarzuZarzu Victor victorzarzu Data 16 decembrie 2021 11:20:31
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.41 kb
#include <bits/stdc++.h>
#define oo 0x3f3f3f3f

using namespace std;
ifstream f("maxflow.in");
ofstream g("maxflow.out");
int n, m;
int cap[1001][1001], fl[1001][1001];
vector<int> graf[1001];
bool viz[1001];
int parent[1001];

void read()
{
  f>>n>>m;
  int x, y;
  for(int i = 1;i <= m;++i) 
    f>>x>>y, f>>cap[x][y], graf[x].push_back(y), graf[y].push_back(x);
}

bool bfs()
{
  memset(viz, false, sizeof(viz));
  queue<int> q;
  q.push(1);

  while(!q.empty())
  {
    int node = q.front();
    q.pop();

    if(node == n) continue;

    for(int vecin : graf[node])
    {
      if(viz[vecin] || cap[node][vecin] == fl[node][vecin]) continue;
      viz[vecin] = true;
      parent[vecin] = node;
      q.push(vecin);
    }
  }

  return viz[n];
}

void solve()
{
  int max_flow = 0;
  while(bfs())
  {
    for(int vecin : graf[n])
    {
      if(!viz[vecin] || cap[vecin][n] == fl[vecin][n]) continue;

      int flow = oo;
      parent[n] = vecin;

      for(int node = n;node != 1;node = parent[node])
        flow = min(flow, cap[parent[node]][node] - fl[parent[node]][node]);
  
      if(!flow)
        continue;
      
      for(int node = n;node != 1;node = parent[node])
        fl[parent[node]][node] += flow, fl[node][parent[node]] -= flow;
      
      max_flow += flow;
    }
  }
  g<<max_flow;
}

int main()
{
  read();
  solve();
  return 0;
}