Cod sursa(job #3220730)

Utilizator diana_dd03Dorneanu Diana diana_dd03 Data 4 aprilie 2024 18:33:34
Problema Flux maxim de cost minim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.83 kb
#include <bits/stdc++.h>
using namespace std;

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

int n, m, source, dest;
int cap[355][355], cost[355][355];
vector <int> v[355], dist;
vector <int> tata;
vector <bool>viz;

int Dijkstra(int start, int stop)
{
    viz.assign(n + 1, 0);
    dist.assign (n + 1, INT_MAX);
    queue <int> coada;
    coada.push (start);
    viz[start] = 1;
    dist[start] = 0;
    tata[start] = start;
    while (!coada.empty())
    {
        int nod = coada.front ();
        coada.pop ();
        viz[nod] = 0;
        for (auto vecin : v[nod])
        {

            if (cap[nod][vecin] > 0 && dist[vecin] > dist[nod] + cost[nod][vecin])
            {
                dist[vecin] = dist[nod] + cost[nod][vecin];
                tata[vecin] = nod;
                if (viz[vecin] == 0)
                {
                    coada.push (vecin);
                    viz[vecin] = 1;
                }
            }
        }
    }
    return dist[stop];
}

int main()
{
    int x , y , a , b;
    fin >> n >> m >> source >> dest;
    for (int i = 1 ; i <= m ; i++)
    {
        fin >> x >> y >> a >> b;
        v[x].push_back (y);
        v[y].push_back (x);
        cap[x][y] = a;
        cost[x][y] = b;
        cost[y][x] = -b;
    }
    tata.assign (n + 1, 0);
    int maxflow = 0;

    while (true)
    {
        int suma = Dijkstra(source, dest);
        if (suma == INT_MAX) break;
        int flow = INT_MAX;
        for (int j = dest; j != source; j = tata[j])
            flow = min (flow, cap[tata[j]][j]);

        for (int j = dest; j != source; j = tata[j])
        {
            cap[tata[j]][j] -= flow;
            cap[j][tata[j]] += flow;
        }
        maxflow += flow * suma;
    }
    fout << maxflow << "\n";
    fout.close();
    return 0;
}