Cod sursa(job #3179132)

Utilizator smunteanuMunteanu Stefan Catalin smunteanu Data 3 decembrie 2023 01:54:19
Problema Flux maxim de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.78 kb
#include <bits/stdc++.h>
using namespace std;


const int NMAX = 357;


int n, m, s, d;
vector<int> adj[NMAX];
int flow[NMAX][NMAX];
int cost[NMAX][NMAX];
int head[NMAX][NMAX];
int par[NMAX];
int dist[NMAX];

bool bellman_ford() {

  static bool inq[NMAX];
  fill(dist + 1, dist + n + 1, INT_MAX);
  fill(inq + 1, inq + n + 1, false);
  fill(par + 1, par + n + 1, -1);

  queue<int> q;

  q.push(s);
  dist[s] = 0;

  while (!q.empty()) {
    int u = q.front();
    q.pop(), inq[u] = false;
    for (int v : adj[u]) {
      if (dist[u] + cost[u][v] < dist[v] && flow[u][v] > 0) {
        dist[v] = dist[u] + cost[u][v];
        par[v] = u;
        if (v != d && !inq[v]) q.push(v), inq[v] = true;
      }
    }
  }

  if (par[d] == -1) return true;

  int fmin = INT_MAX;

  for (int u = n; u != s; u = par[u]) {
    fmin = min(fmin, flow[par[u]][u]);
  }

  for (int u = n; u != s; u = par[u]) {
    flow[par[u]][u] -= fmin;
    flow[u][par[u]] += fmin;
  }

  return false;
}


void solve() {

  cin >> n >> m >> s >> d;

  for (int i = 0; i < m; i++) {
    int u, v, f, c;
    cin >> u >> v >> f >> c;
    adj[u].push_back(v);
    adj[v].push_back(u);
    head[u][v] = u;
    head[v][u] = u;
    flow[u][v] = f;
    cost[u][v] = c;
    cost[v][u] = -c;
  }

  while (!bellman_ford());

  long long res = 0;

  for (int u = 1; u <= n; u++) {
    for (int v = u + 1; v <= n; v++) {
      res += 1ll * flow[u+v-head[u][v]][head[u][v]] * cost[head[u][v]][u+v-head[u][v]];
    }
  }

  cout << res << endl;

}


int main() {

  #ifdef LOCAL
  freopen("file.in", "r", stdin);
  #else
  freopen("fmcm.in", "r", stdin);
  freopen("fmcm.out", "w", stdout);
  #endif

  ios_base::sync_with_stdio(false), cin.tie(NULL);

  solve();
}