Cod sursa(job #2085575)

Utilizator MiricaMateiMirica Matei MiricaMatei Data 10 decembrie 2017 13:45:04
Problema Flux maxim de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.73 kb
#include <bits/stdc++.h>

using namespace std;

const int INF = (1LL << 31) - 1;
const int MAX_N = 350;

typedef pair<int, int> p;

vector<int>G[MAX_N + 5];
int nrv[MAX_N + 5];
bool inc[MAX_N + 5];
int dist[MAX_N + 5], realdist[MAX_N + 5], dist2[MAX_N + 5];
int from[MAX_N + 5];
bool muchie[MAX_N + 5][MAX_N + 5];
int cr[MAX_N + 5][MAX_N + 5];
int cost[MAX_N + 5][MAX_N + 5];
int cost1[MAX_N + 5][MAX_N + 5];

int dijkstra(int s, int d, int n) {
  for (int i = 1; i <= n; ++i)
    dist2[i] = INF;
  dist2[s] = 0;
  realdist[s] = 0;

  priority_queue<p, vector<p>,greater<p> > pq;
  pair<int, int>st = make_pair(0, s);
  pq.push(st);
  while (!pq.empty()) {
    p aux = pq.top();
    pq.pop();
    if (dist2[aux.second] == aux.first) {
      for (auto it:G[aux.second]) {
        int val = dist[aux.second] + cost[aux.second][it] - dist[it];
        if (cr[aux.second][it] > 0 && aux.first + val < dist2[it]) {
          dist2[it] = aux.first + val;
          from[it] = aux.second;
          realdist[it] = realdist[aux.second] + cost[aux.second][it];
          p aux1 = make_pair(dist2[it], it);
          pq.push(aux1);
        }
      }
    }
  }
  for (int i = 1; i <= n; ++i)
    dist[i] = realdist[i];
  return dist[d];
}

int fmcm(int s, int d, int n) {
  int ans = 0, dm;
  dm = dijkstra(s, d, n);
  while (dist2[d] != INF) {
    int aux = INF;
    int node = d;
    while (node != s) {
      aux = min(aux, cr[from[node]][node]);
      node = from[node];
    }
    node = d;
    while (node != s) {
      cr[from[node]][node] -= aux;
      cr[node][from[node]] += aux;
      node = from[node];
    }
    ans += dm * aux;
    dm = dijkstra(s, d, n);
  }
  return ans;
}

int main() {
  freopen("fmcm.in", "r", stdin);
  freopen("fmcm.out", "w", stdout);

  int n, m, s, d;
  scanf("%d%d%d%d", &n, &m, &s, &d);
  for (int i = 1; i <= m; ++i) {
    int u, v, c, cst;
    scanf("%d%d%d%d", &u, &v, &c, &cst);
    G[u].push_back(v);
    G[v].push_back(u);
    cr[u][v] = c;
    muchie[u][v] = 1;
    cost[u][v] = cst;
    cost[v][u] = -cst;
  }

  for (int i = 1; i <= n; ++i) {
    dist[i] = INF;
    nrv[i] = 0;
    inc[i] = 0;
  }
  inc[s] = 1;
  from[s] = s;
  nrv[s] = 1;
  dist[s] = 0;
  queue<int>q;
  q.push(s);
  while (!q.empty() && nrv[q.front()] < n) {
    int node = q.front();
    q.pop();
    inc[node] = 0;
    for (auto it:G[node]) {
      if (cr[node][it] > 0 && dist[node] + cost[node][it] < dist[it]) {
        from[it] = node;
        dist[it] = dist[node] + cost[node][it];
        if (inc[it] == 0) {
          inc[it] = 1;
          q.push(it);
          nrv[it]++;
        }
      }
    }
  }

  printf("%d\n", fmcm(s, d, n));

  return 0;
}