Cod sursa(job #3267988)

Utilizator Bogdan345Marius Mihalache Bogdan345 Data 13 ianuarie 2025 12:53:13
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.17 kb
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
ifstream cin("maxflow.in");
ofstream cout("maxflow.out");
queue<int>q;
vector<vector<int>>flux,gr,capacitate;
vector<int>tata;

int n,m,nod1,nod2,cap,S,T,rasp=0;
const int inf=1e9;

bool bfs(){
    for(int i=1;i<=n;i++){
        tata[i]=0;
    }
    tata[S]=S;
    q.push(S);

    bool ok=false;

    while(!q.empty()){

        int nod_curent=q.front();
        q.pop();

        if(nod_curent==T){
            ok=true;//am gasit cel putin un drum nesaturat 
            continue;
        }
        for(auto&vecin:gr[nod_curent]){

            if(!tata[vecin] && capacitate[nod_curent][vecin]-flux[nod_curent][vecin]>0){
                tata[vecin]=nod_curent;
                q.push(vecin);
            }
        }
    }
    return ok;
}
int main(){

    cin>>n>>m;

    S=1;
    T=n;

    gr.resize(n+1);
    tata.resize(n+1,0);
    capacitate.resize(n+1,vector<int>(n+1));
    flux.resize(n+1,vector<int>(n+1));

    for(int i=1;i<=m;i++){
        cin>>nod1>>nod2>>cap;

        gr[nod1].push_back(nod2);
        gr[nod2].push_back(nod1);
        capacitate[nod1][nod2]=cap;
    }

    while(bfs()){//cat timp exista lant nesaturat de la s la t,verific daca mai pot trimite flux,respectiv daca pot redirectiona
        for(auto& vecin:gr[T]){//ma uit in toate nodurile care sunt adiacente cu destinatia,deoarece prin ele 
        //vor trece drumurile cu fluxul adaugat
             if(!tata[vecin]){
                continue;
             }
             tata[T]=vecin;
             int nod_curent=T;
             int minn=inf;

             while(nod_curent!=S){
                minn=min(minn,capacitate[tata[nod_curent]][nod_curent]-flux[tata[nod_curent]][nod_curent]);
                nod_curent=tata[nod_curent];
             }
             nod_curent=T;

             while(nod_curent!=S){
                flux[tata[nod_curent]][nod_curent]+=minn;
                flux[nod_curent][tata[nod_curent]]-=minn;
                nod_curent=tata[nod_curent];
             }
             rasp+=minn;
        }        
    }
    cout<<rasp<<'\n';
    return 0;
}