Untitled

 avatar
unknown
plain_text
a year ago
3.9 kB
3
Indexable

#include <iostream>
#include <fstream>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <cstdlib>
#include <cstdio>
#include <string>
#include <cstring>
#include <cassert>
#include <utility>
#include <iomanip>

using namespace std;

const int MAXN = 1050;
const long long INF = (long long) 1e15;

struct edge {
    int from, to;
    int flow, cap;
    long long cost;
};

int n;
int cost[MAXN][MAXN];
vector <edge> e;
vector <int> g[MAXN];
long long phi[MAXN];
priority_queue < pair < long long, int > > q;
long long dist[MAXN];
int par[MAXN];
int edge_num;
int s = 0, t = MAXN - 1;

void fordBellman() {
    for (int i = s; i <= t; i++) 
        dist[i] = INF;
    dist[s] = 0;
    while (true) {
        bool change = false;
        for (int i = 0; i < edge_num; i++) {
            int from = e[i].from, to = e[i].to;
            if (e[i].flow == e[i].cap)
                continue;
            if (dist[from] == INF)
                continue;
            if (dist[to] > dist[from] + e[i].cost) {
                dist[to] = dist[from] + e[i].cost;
                change = true;
            }
        }
        if (!change)
            break;
    }
}

void dijkstra() {
    while (!q.empty())
        q.pop();
    for (int i = s; i <= t; i++) {
        dist[i] = INF;
        q.push(make_pair(-dist[i], i));
    }
    dist[s] = 0;
    q.push(make_pair(0, s));
    while (!q.empty()) {
        int cur = q.top().second;
        long long cur_dist = -q.top().first;
        q.pop();
        if (cur_dist > dist[cur])
            continue;
        if (dist[cur] == INF)
            break;
        for (int i = 0; i < (int) g[cur].size(); i++) {
            int ind = g[cur][i];
            if (e[ind].flow == e[ind].cap)
                continue;
            int to = e[ind].to;
            long long w = e[ind].cost + phi[cur] - phi[to];
            if (cur_dist + w < dist[to]) {
                dist[to] = cur_dist + w;
                par[to] = ind;
                q.push(make_pair(-dist[to], to));
            }
        }
    }
}

long long minCost(int flow) {
    long long result = 0;

    fordBellman();  
    for (int i = s; i <= t; i++)
        phi[i] = dist[i];

    while (true) {

        dijkstra();

        if (dist[t] == INF)
            return result;

        for (int i = s; i <= t; i++)
            phi[i] = min(phi[i] + dist[i], INF);

        int push = flow;
        int cur = t;
        while (cur != s) {
            edge tmp = e[par[cur]];
            int from = tmp.from, can_push = tmp.cap - tmp.flow;
            push = min(push, can_push);
            cur = from;
        }

        flow -= push;
        cur = t;
        while (cur != s) {
            edge tmp = e[par[cur]];
            int from = tmp.from;
            e[par[cur]].flow += push;
            e[par[cur] ^ 1].flow -= push;
            result += 1ll * push * tmp.cost;
            cur = from;
        }

        if (flow == 0)
            break;
    }
    return result;
}   

void addEdge(int from, int to, int cap, long long cost) {
    edge tmp;
    tmp.from = from; tmp.to = to; tmp.flow = 0; tmp.cap = cap; tmp.cost = cost;
    e.push_back(tmp);
    g[from].push_back(edge_num);

    tmp.from = to; tmp.to = from; tmp.flow = cap; tmp.cap = cap; tmp.cost = -cost;
    e.push_back(tmp);
    g[to].push_back(edge_num + 1);

    edge_num += 2;
}


int main() {
  std::ios::sync_with_stdio(false);
  std::cin.tie(0);

  int N, M;
  std::cin >> N;
  std::cin >> M;


    int from, to;
    int64_t capacity, c;


n = N;
s = 0;
t = N - 1;

    for (int i = 0; i < M; ++i) {
        std::cin >> from;
        std::cin >> to;
        std::cin >> capacity;
        std::cin >> c;

        addEdge(from - 1, to - 1, capacity, c);
        cost[from - 1][to - 1] = c;
    }


    cout << minCost(n) << endl;

  return 0;
}
Editor is loading...
Leave a Comment