Cod sursa(job #2807874)

Utilizator vladth11Vlad Haivas vladth11 Data 24 noiembrie 2021 12:12:19
Problema Flux maxim de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.52 kb
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define debug(x) cerr << #x << " " << x << "\n"
#define debugs(x) cerr << #x << " " << x << " "

using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair <ll, ll> pii;
typedef pair <long double, pii> muchie;

const ll NMAX = 351;
const ll VMAX = 21;
const ll INF = (1LL << 55);
const ll MOD = 998244353;
const ll BLOCK = 318;
const ll base = 31;
const ll nr_of_bits = 30;

vector <int> v[NMAX];

struct edge {
    int from, to, cap, cost;
} edges[25001];

int cnt = 0, s, d, n, viz[NMAX];
ll dist[NMAX], last[NMAX], distt[NMAX], distn[NMAX];

void addEdge(int a, int b, int c, int cost) {
    edges[++cnt] = {a, b, c, cost};
    v[a].push_back(cnt);
    edges[++cnt] = {b, a, 0, -cost};
    v[b].push_back(cnt);
}

int invers(int x) {
    if(x % 2)
        return (x + 1);
    return (x - 1);
}

void bellman() {
    for(int i = 1; i <= n; i++) {
        viz[i] = 0, distt[i] = 2e9;
        last[i] = 0;
    }
    queue <int> pq;
    pq.push(s);
    viz[s] = 1;
    distt[s] = 0;
    while(pq.size()) {
        int node = pq.front();
        pq.pop();
        viz[node] = 0;
        for(auto x : v[node]) {
            if(!edges[x].cap)
                continue;
            int to = edges[x].to;
            int cost = edges[x].cost;
            if(distt[node] + cost < distt[to]) {
                distt[to] = distt[node] + cost;
                last[to] = x;
                if(!viz[to]) {
                    pq.push(to);
                    viz[to] = 1;
                }
            }
        }
    }
}

bool Dijkstra(){
    for(ll i = 1; i <= n; i++){
        viz[i] = 0, dist[i] = 2e9, distn[i] = 2e9;
        last[i] = 0;
    }
    priority_queue <pii> pq;
    pq.push({0, s});
    dist[s] = 0;
    distn[s] = 0;
    while(pq.size()){
        while(pq.size() && viz[pq.top().second])
            pq.pop();
        if(!pq.size())
            continue;
        ll node = pq.top().second;
        pq.pop();
        viz[node] = 1;
        for(auto x : v[node]){
            if(!edges[x].cap)
                continue;
            ll to = edges[x].to;
            ll cost = edges[x].cost + distt[node] - distt[to];
            if(dist[node] + cost < dist[to]){
                dist[to] = dist[node] + cost;
                distn[to] = distn[node] + edges[x].cost;
                pq.push({-dist[to], to});
                last[to] = x;
            }
        }
    }
    for(int i = 1; i <= n; i++)
        distt[i] = distn[i];
    if(viz[d])
        return 1;
    return 0;
}

ll flow() {
    ll sol = 0;
    while(Dijkstra()) {
        ll minim = 2e9;
        ll node = last[d];
        while(node != 0) {
            minim = min(minim, 1LL * edges[node].cap);
            node = last[edges[node].from];
        }
        sol += 1LL * minim * distt[d];
        node = last[d];
        while(node != 0) {
            edges[node].cap -= minim;
            edges[invers(node)].cap += minim;
            node = last[edges[node].from];
        }
    }
    return sol;
}

int main() {
    ifstream cin("fmcm.in");
    ofstream cout("fmcm.out");
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int m, i;
    cin >> n >> m >> s >> d;
    for(i = 1; i <= m; i++) {
        int a, b, c, d;
        cin >> a >> b >> c >> d;
        addEdge(a, b, c, d);
    }
    bellman();
    cout << flow();
    return 0;
}