Cod sursa(job #2960887)

Utilizator Rincu_StefaniaRincu Stefania Rincu_Stefania Data 5 ianuarie 2023 10:56:41
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.69 kb
#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
#include <cstring>
#include <fstream>

using namespace std;

ifstream in("maxflow.in");
ofstream out("maxflow.out");

///flow_sent -> fluxul trimis pe o muchie, capacity -> capacitatea muchiei,
///good_flow -> fluxul maxim trimis pe drumul dat de BFS
int flow_sent[1005][1005], capacity[1005][1005], good_flow[1005] = {0};
///l -> vecorul care retine graful + graful rezidual
vector<int> l[1005];
///n -> noduri, m -> muchii
int n, m;
///vectprul de parinti
int tata[1005] = {0}, viz[1005] = {0};

int bfs()
{
    queue<int> q;
    ///rescriem valorile din vectorul de tati si cel de vizitate
    memset(tata, 0, sizeof(tata));
    memset(viz, 0, sizeof(viz));

    q.push(1);
    viz[1] = 1;

    ///BFS clasic
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        ///iesim cu 1 daca am ajuns la nodul n(cel final)
        if(u == n)
            return 1;
        for(int i = 0; i < l[u].size(); i++)
        {
            int v = l[u][i];
            ///inseamna ca nu e vizitat
            if(capacity[u][v] > 0 && !viz[v])
            {
                ///updatam vectorul de tati pentru a avea drumul
                tata[v] = u;
                ///setam nodul ca vizitat
                viz[v] = 1;
                q.push(v);
            }
        }
    }
    return 0;
}

///pentru a optimiza programul efectuam algoritmul lui Edmonds Karp atata timp
///cat prin parcurerea BFS putem ajunge din nodul 1 in nodul n, ceea ce inseamna
///ca estista un drum pe care fluxul rezidual sa fie minim si mai am capacitate
///disponibila pe muchii pentru a mari fluxul
int edmonds_karp()
{
    ///maxim -> fluxul maxim, minim -> fluxul minim rezidual gasit dupa parcurgerea BFS
    ///u -> nodul curent, v -> vecinul acestuia
    int maxim = 0, minim, u, v;
    ///cat timp prin parcurgerea BFS ajung din nodul 1 in nodul n
    while(bfs())
    {
        ///iterez prin vecinii nodului n(final)
        for(int i = 0; i < l[n].size(); i++)
        {
            u = l[n][i];
            ///daca mai am capacitate disponibila pe muchia dintre noduri
            ///si nodul e vizitat atunci calculam fluxul minim rezidual
            if(capacity[u][n] > 0 && viz[u])
            {
                ///in BFS am pastrat in vectorul de tati drumul gasit pentru transmiterea fluxului
                ///iar acum parcurgem acel drum pentru a gasi minimul si pentru a reactualiza capacitatile
                ///de pe muchii, in functie de cum apar in parcurgere

                minim = capacity[u][n];
                while(tata[u])
                {
                    v = tata[u];
                    minim = min(minim, capacity[v][u]);
                    u = v;
                }
                maxim += minim;
                u = l[n][i];
                capacity[u][n] -= minim;
                capacity[n][u] += minim;
                while(tata[u])
                {
                    v = tata[u];
                    capacity[v][u] -= minim;
                    capacity[u][v] += minim;
                    u = v;
                }
            }
        }
    }
    return maxim;
}

int main()
{
    int s, f, cap, max_flow;
    in>>n>>m;
    while(m)
    {
        in>>s>>f>>cap;
        ///punem capacitatea de la nodul de start la cel de final
        capacity[s][f] = cap;
        ///graful propriu-zis
        l[s].push_back(f);
        ///graful rezidual
        l[f].push_back(s);
        m--;
    }

    ///calculam fluxul maxim cu ajutorul algoritmului Edmonds Karp
    max_flow = edmonds_karp();

    out<<max_flow<<"\n";

    return 0;
}