Cod sursa(job #2357914)

Utilizator DavidLDavid Lauran DavidL Data 27 februarie 2019 20:05:56
Problema Flux maxim de cost minim Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.79 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fi("fmcm.in");
ofstream fo("fmcm.out");

const int NMAX = 400;
const int MMAX = 13000;
const int INF = 5e8;

int n, m, s, d;
int C[NMAX][NMAX], F[NMAX][NMAX];
int cost[NMAX][NMAX];
vector <int> G[NMAX];
int p[NMAX];
int dist[NMAX];
bool inCoada[NMAX];

bool drum()
{
    memset(p, 0, sizeof(p));
    memset(inCoada, 0, sizeof(inCoada));
    for (int i = 1; i <= n; i++)
        dist[i] = INF;

    queue <int> Q;
    Q.push(s);
    dist[s] = 0;
    inCoada[s] = 1;

    while (!Q.empty())
    {
        int nod = Q.front();
        Q.pop();
        inCoada[nod] = 0;

        for (auto v: G[nod])
        {
            if (dist[nod] + cost[nod][v] < dist[v] && F[nod][v] != C[nod][v] && !inCoada[v]) /// drum mai bun si nu e saturata muchia
            {
                dist[v] = dist[nod] + cost[nod][v];
                Q.push(v);
                inCoada[v] = 1;
                
                p[v] = nod;
                //if (v == d)
                    //return 1;
            }
        }
    }
    return dist[d] != INF;
}

int main()
{
    fi >> n >> m >> s >> d;
    for (int i = 1; i <= m; i++)
    {
        int x, y, c, z;
        fi >> x >> y >> c >> z;
        G[x].push_back(y);
        G[y].push_back(x);
        C[x][y] = c;
        C[y][x] = 0;
        cost[x][y] = z;
        cost[y][x] = -z;
    }

    int rez = 0;
    while (drum())
    {
        int minim = INF;
        for (int i = d; i != s; i = p[i])
            minim = min(minim, C[p[i]][i] - F[p[i]][i]);

        for (int i = d; i != s; i = p[i])
        {
            F[p[i]][i] += minim;
            F[i][p[i]] -= minim;

            rez += cost[p[i]][i] * minim;
        }
    }
    fo << rez;

    return 0;
}