Cod sursa(job #2956639)

Utilizator allee69Aldea Alexia allee69 Data 19 decembrie 2022 23:01:39
Problema Flux maxim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.47 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#define INF 110001

using namespace std;

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

int n, m;
int capacitate[1001][1001];
vector<int>tata, viz;
vector<int>graf[1001];
queue<int>q;

int BF()
{
    viz.assign(n+1, 0);
    tata.assign(n+1, 0);
    while(!q.empty())
        q.pop();

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

    while(!q.empty())
    {
        int nod=q.front();
        q.pop();

        if(nod==n) continue;

        for(auto j:graf[nod])
        {
            if(viz[j]==0 && capacitate[nod][j]>0)
            {
                q.push(j);
                tata[j]=nod;
                viz[j]=1;
            }
        }
    }
    return viz[n];
}

int main()
{
    f>>n>>m;
    int x,y,c,flowmaxim;
    for(int i=1; i<=m; i++)
    {
        f>>x>>y>>c;
        capacitate[x][y] = c;
        graf[x].push_back(y);
        graf[y].push_back(x);
    }
    while(BF())
    {
        for(auto nod: graf[n])
        {
            if(!viz[nod]) continue;

            tata[n]=nod;
            int flow=INF;

            for(nod=n; nod!=1; nod=tata[nod])
                flow=min(flow, capacitate[tata[nod]][nod]);

            if(flow==0) continue;

            for(nod=n; nod!=1; nod=tata[nod])
            {
                capacitate[tata[nod]][nod]-=flow;
                capacitate[nod][tata[nod]]+=flow;
            }
            flowmaxim+=flow;
        }

    }
    g<<flowmaxim;
    return 0;
}