Cod sursa(job #2124559)

Utilizator alexilasiAlex Ilasi alexilasi Data 7 februarie 2018 12:54:20
Problema Flux maxim de cost minim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.82 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>

using namespace std;

ifstream fin("fmcm.in");
ofstream fout("fmcm.out");

#define nmax 355
#define inf 1000000000
#define mp make_pair
#define f first
#define s second

vector <int> v[nmax];
int c[nmax][nmax];
int f[nmax][nmax];
int cst[nmax][nmax];
int t[nmax];
int viz[nmax];
int d[nmax];
int n,a,b,m;

bool dijkstra()
{
    for(int i=1; i<=n; ++i)
        d[i]=inf;
    priority_queue<pair<int,int>, vector<pair<int,int> >, greater <pair<int,int> > >h;
    h.push(mp(0,a));
    d[a]=0;
    while(h.size())
    {
        int nod = h.top().s, cost = h.top().f;
        h.pop();
        if(nod == b || cost!=d[nod])
            continue;
        for(auto it:v[nod])
        {
            if(f[nod][it]<c[nod][it] && d[nod]+cst[nod][it]<d[it])
            {
                d[it]=d[nod]+cst[nod][it];
                t[it]=nod;
                h.push(mp(d[it],it));
            }
        }
    }
    return d[b]!=inf;
}

int main()
{
    fin>>n>>m>>a>>b;
    while(m--)
    {
        int x,y,q,z;
        fin>>x>>y>>q>>z;
        c[x][y]=q;
        cst[x][y]=z;
        cst[y][x]=-z;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    int ans = 0;
    while(dijkstra())
    {
        for(auto it:v[b])
        {
            if(d[it]==inf || c[it][b]==f[it][b])
                continue;
            int flow = inf;
            for(int i = b; i!=a; i=t[i])
                flow = min(flow,c[t[i]][i]-f[t[i]][i]);
            if(!flow)
                continue;
            for(int i = b; i!=a; i=t[i])
            {
                f[t[i]][i]+=flow;
                f[i][t[i]]-=flow;
                ans+=flow*cst[t[i]][i];
            }
        }
    }
    fout<<ans;
}