Cod sursa(job #2691444)

Utilizator miramaria27Stroie Mira miramaria27 Data 28 decembrie 2020 17:58:22
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.21 kb
#include <iostream>
#include <queue>
#include <fstream>
#include <algorithm>
#include <utility>

using namespace std;

ifstream fin("maxflow.in");
ofstream fout("maxflow.out");
///o sa implementez a doua varianta prezentata la curs - gasesc fluxul
/// maxim folosindu-ma de graful rezidual si retin arborele bfs cu ajutorul
///unui array de tati
/// de asemenea, o sa retinem  graful folosindu-ne de o matrice de adiacenta
/// pentru care g[i][j] = capacitatea arcului cu extremitatiile (i,j) daca (i,j) apartine lui E
//                        0, altfel

int inf = 100000;
vector<int> tata (1002,0),viz(1002,0);
int n, m, s, t;
int gr[1002][1002];
vector <pair<int,int>> ad[1002];
queue<int> q;
///o sa fac un bfs care sa imi intoarca true daca a fost
///gasit un path de la s la t si false altfel
bool bfs()
{
    bool found = false;
    while(!q.empty())
    {
        int nod = q.front();
        if(nod != t)
            for(auto &j: ad[nod])
            {
                int i = j.first;
                if(gr[nod][i] && viz[i] == 0)
                {
                    q.push(i);

                    if( i == t)
                        found = true;
                    viz[i] = 1;
                    tata[i] = nod;
                }
            }
        q.pop();
    }
    return found;

}
int main()
{
    int flux = 0;
    fin >> n >> m;
    s = 1;
    t = n;
    for(int i  = 1; i <=m ; i++)
    {
        int a,b,c;
        fin>>a>>b>>c;
        ad[a].push_back(make_pair(b,c));
        ad[b].push_back(make_pair(a,c));
        ///in graful rezidual semnalam ca putem sa trimitem c "obiecte"
        /// de la a la b
        gr[a][b] = c;
    }

    q.push(s);
    while(bfs())
    {

        ///pentru toate varfurile x, adiacente  cu varful destinatie t, care stiu ca apartin arborelui bfs, si pentru care stiu
        ///ca : 1. exista un arc (x,t) in graful rezidual (adica pot sa transport obiecte de la x la t de o anumita capacitate)
        ///     2. f(xt) != capacitate (xt) (inca mai am loc sa transport obiecte)
        for(int j = 0; j < ad[t].size(); j++)
        {


            int i = ad[n][j].first;
            int cap = ad[n][j].second;
            if(gr[i][n] && viz[i] == 1 && gr[n][i] != cap)
            {


                ///vream sa computam i(P)  - capacitatea reziduala a drumului P in graful rezidual

                int nod = i;
                int capacity = gr[i][n];
                while(nod != s)
                {
                    capacity = min(capacity,gr[tata[nod]][nod]);
                    nod = tata[nod];

                }

                ///revizuim drumul si actualizam graful rezidual
                nod = i;
                gr[nod][n] -= capacity;
                gr[n][nod] += capacity;
                while(nod != s)
                {

                    gr[tata[nod]][nod] -= capacity;
                    gr[nod][tata[nod]] += capacity;
                    nod = tata[nod];
                }
                flux += capacity;
            }
        }
        q.push(s);
        fill(tata.begin(),tata.end(),0);
        fill(viz.begin(),viz.end(),0);
    }
    fout<< flux;
    return 0;
}