Cod sursa(job #1940570)

Utilizator vladm98Munteanu Vlad vladm98 Data 26 martie 2017 17:57:58
Problema Flux maxim de cost minim Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 2.05 kb
#include <bits/stdc++.h>

using namespace std;
int flux[1001][1001], capacitate[1001][1001];
deque <int> sol;
int tata[1001];
int s, d;
int viz[1001];
int dist[1001];
int n;
vector <pair <int, int>> graf[1001];
bool maxflow (int nod)
{
    dist[nod] = 0;
    sol.push_back(nod);
    viz[nod] = 1;
    int verif = 0;
    while (!sol.empty())
    {
        int node = sol.front();
        sol.pop_front();
        viz[node] = 0;
        if (node == d)
            verif = 1;
        for (auto x:graf[node])
        {
            if (capacitate[node][x.first]>flux[node][x.first] && dist[x.first]>dist[node]+x.second)
            {
                tata[x.first] = node;
                dist[x.first] = dist[node]+x.second;
                if (viz[x.first] == 0)
                    viz[x.first]=1, sol.push_back(x.first);
            }
        }
    }
    return verif;
}
int main()
{
    ifstream fin ("fmcm.in");
    ofstream fout ("fmcm.out");
    int m, fluxul = 0;
    fin >> n >> m >> s >> d;
    for (int i = 1; i<=m; ++i)
    {
        int x, y, cap, cost;
        fin >> x >> y >> cap >> cost;
        graf[x].push_back(make_pair(y, cost));
        graf[y].push_back(make_pair(x, -cost));
        capacitate[x][y] += cap;
    }
    memset(dist, 1<<30, sizeof(dist));
    for (; maxflow(s);)
    {
        int minim = 1<<30;
        int copie = d;
        while (copie!=s)
        {
            if (minim > capacitate[tata[copie]][copie]-flux[tata[copie]][copie])
                minim = capacitate[tata[copie]][copie]-flux[tata[copie]][copie];
            copie = tata[copie];
        }
        if (minim == 0)
            continue;
        fluxul+=(minim*dist[d]);
        copie = d;
        while (copie!=s)
        {
            flux[tata[copie]][copie]+=minim;
            flux[copie][tata[copie]]-=minim;
            copie = tata[copie];
        }
        memset(tata, 0, sizeof(tata));
        memset(dist, 1<<30, sizeof(dist));
        memset(viz, 0, sizeof(viz));
    }
    fout << fluxul;
    return 0;
}