Cod sursa(job #2778432)

Utilizator SochuDarabaneanu Liviu Eugen Sochu Data 1 octombrie 2021 13:14:45
Problema Flux maxim de cost minim Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.47 kb
#include<bits/stdc++.h>
#pragma GCC oprimize ("03")
#define FastIO ios_base::sync_with_stdio(false) , cin.tie(0) , cout.tie(0)
#define FILES freopen("fmcm.in" , "r" , stdin) , freopen("fmcm.out" , "w" , stdout)
#define pii pair < int , int >
#define pb push_back

using namespace std;

const int N = 355;

int n , m , source , sink , x , y , z , c;
int capacity[N][N] , weight[N][N] , parent[N];
int d[N] , nd[N];
vector < int > G[N];
priority_queue < int , vector < pii > , greater < pii > > s;
queue < int > q;
bool viz[N];

void bellman()
{
    int i;

    for(i = 1 ; i <= n ; i++)
        d[i] = INT_MAX;

    d[source] = 0;
    q.push(source);

    while(!q.empty())
    {
        int u = q.front();
        q.pop();

        for(auto v : G[u])
        {
            if(capacity[u][v] && d[v] > d[u] + weight[u][v])
            {
                d[v] = d[u] + weight[u][v];
                q.push(v);
            }
        }
    }
}

int k = INT_MAX;
int flow;
int minCost = 0;

bool inline djikstra()
{
    int i;

    for(i = 1 ; i <= n ; i++)
        nd[i] = INT_MAX;

    nd[source] = 0;
    s.push({0 , source});

    while(!s.empty())
    {
        int u = s.top().second;
        s.pop();

        for(auto v : G[u])
        {
            if(capacity[u][v])
            {
                if(nd[v] > nd[u] + weight[u][v] + d[u] - d[v])
                {
                    parent[v] = u;
                    nd[v] = nd[u] + weight[u][v] + d[u] - d[v];
                    s.push({nd[v] , v});
                }
            }
        }
    }

    if(nd[sink] == INT_MAX)
        return 0;

    int minFlow = k - flow;
    int curr;

    for(curr = sink ; curr != source ; curr = parent[curr])
        minFlow = min(minFlow , capacity[parent[curr]][curr]);

    for(curr = sink ; curr != source ; curr = parent[curr])
    {
        capacity[parent[curr]][curr] -= minFlow;
        capacity[curr][parent[curr]] += minFlow;
        minCost += minFlow * weight[parent[curr]][curr];
    }

    flow += minFlow;

    return 1;
}

signed main()
{
	#ifndef ONLINE_JUDGE
		FastIO , FILES;
	#endif

    cin >> n >> m >> source >> sink;

    while(m--)
    {
        cin >> x >> y >> c >> z;

        capacity[x][y] = c;

        weight[x][y] = z;
        weight[y][x] = -z;

        G[x].pb(y);
        G[y].pb(x);
    }

    bellman();
    for( ; djikstra() ; );
    cout << minCost;

    return 0;
}