Cod sursa(job #3156216)

Utilizator lolismekAlex Jerpelea lolismek Data 10 octombrie 2023 21:04:25
Problema Flux maxim de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.99 kb
#include <algorithm>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <vector>
#include <queue>
#include <stack>

using namespace std;

#define int long long
#define pii pair <int, int>

string filename = "fmcm";

#ifdef LOCAL
    ifstream fin("input.in");
    ofstream fout("output.out");
#else
    ifstream fin(filename + ".in");
    ofstream fout(filename + ".out");
#endif

const int NMAX = 350;
const int INF = 1e9;

int f[NMAX + 1][NMAX + 1];

vector <pii> adj[NMAX + 1];

int aux[NMAX + 1];
bool inQ[NMAX + 1];

int n, m, s, d;

void BF(){
    for(int i = 1; i <= n; i++){
        aux[i] = INF;
    }

    queue <int> Q;
    Q.push(s);
    aux[s] = 0;
    inQ[s] = true;

    while(!Q.empty()){
        int node = Q.front();
        inQ[node] = false;
        Q.pop();

        for(pii vec : adj[node]){
            if(f[node][vec.first] > 0 && vec.second + aux[node] < aux[vec.first]){
                aux[vec.first] = aux[node] + vec.second;
                if(!inQ[vec.first]){
                    Q.push(vec.first);
                    inQ[vec.first] = true;
                }
            }
        }
    }
}

int par[NMAX + 1];
int dist[NMAX + 1];
int real[NMAX + 1];

bool DJK(){
    for(int i = 1; i <= n; i++){
        par[i] = 0;
        dist[i] = real[i] = INF;
    }

    priority_queue <pii, vector <pii>, greater <pii>> pq;
    dist[s] = real[s] = 0;
    pq.push({dist[s], s});

    while(!pq.empty()){
        int node = pq.top().second, val = pq.top().first;
        pq.pop();

        if(val == dist[node]){
            for(pii vec : adj[node]){
                if(f[node][vec.first] > 0 && dist[node] + aux[node] - aux[vec.first] + vec.second < dist[vec.first]){
                    dist[vec.first] = dist[node] + aux[node] - aux[vec.first] + vec.second;
                    real[vec.first] = real[node] + vec.second;
                    par[vec.first] = node;
                    pq.push({dist[vec.first], vec.first});
                }
            }
        }
    }

    return (dist[d] != INF);
}

signed main(){

    fin >> n >> m >> s >> d;

    for(int i = 1; i <= m; i++){
        int x, y, c, z;
        fin >> x >> y >> c >> z;
        adj[x].push_back({y, +z});
        adj[y].push_back({x, -z});
        f[x][y] = c;
    }

    BF();

    int maxflow = 0, mincost = 0;

    cout << 1;

    while(DJK()){
        int node = d, mini = INF;

        while(node != s){
            mini = min(f[par[node]][node], mini);
            node = par[node];
        }

        if(mini == 0){
            continue;
        }

        node = d;
        while(node != s){
            f[par[node]][node] -= mini;
            f[node][par[node]] += mini;
            node = par[node];
        }

        maxflow += mini;
        mincost += mini * real[d];

        for(int i = 1; i <= n; i++){
            aux[i] = dist[i];
        }
    }

    fout << mincost << '\n';

    return 0;
}