Cod sursa(job #3293827)

Utilizator Sorin_GabrielGabara Sorin Gabriel Sorin_Gabriel Data 12 aprilie 2025 17:55:05
Problema Flux maxim Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.28 kb
#include <bits/stdc++.h>
#define INF 2000000000
#define VMAX 1005
#define NMAX 5005
using namespace std;
ifstream fin ("maxflow.in");
ofstream fout ("maxflow.out");

int flux_pe_muchie[NMAX];
bool vizitat[VMAX];
queue<int> noduri_vizitate_in_parcurgere;
int n;

vector<int> graf[VMAX];
// cat flux merge prin muchie si limita


int capacitate[VMAX][VMAX];
int flux[VMAX][VMAX];


int dfs(int nod, int minim) // minim repr cat flux avem de distribuit prin muchiile din nodul nod
{
    int flux_total = 0;
    if(nod == n) // am ajuns la sink
        return minim;
    vizitat[nod]=1;
    noduri_vizitate_in_parcurgere.push(nod);


    for(auto i:graf[nod])
    {
        if(vizitat[i])
            continue;
        if(capacitate[nod][i]==0 && flux[nod][i]>0) // muchie inversa si avem de unde intoarce flux
        {
            int nr = dfs(i,min(minim,flux[nod][i])); // cat flux a incaput prin acel path
            if(nr)
            {
                flux[nod][i]-=nr;
                flux[i][nod]+=nr;
                flux_total+=nr;
                minim-=nr;
                //vizitat[nod]=0;
                //return nr;
            }
        }
        else if(capacitate[nod][i]>0 && capacitate[nod][i]-flux[nod][i]>0) // muchie normala si avem unde adauga flux
        {
            int nr = dfs(i,min(minim,capacitate[nod][i]-flux[nod][i])); // cat flux a incaput prin acel path
            if(nr)
            {
                flux[nod][i]+=nr;
                flux[i][nod]-=nr;
                flux_total+=nr;
                minim-=nr;
                //vizitat[nod]=0;
                //return nr;
            }
        }
    }

    //return 0;
    //vizitat[nod]=0;
    return flux_total;
}


int main()
{
    long long int m,i,j,k,t,q,nr,suma,max_flow;
    fin>>n>>m;
    for(i=0;i<m;i++)
    {
        fin>>j>>k>>t;
        capacitate[j][k]+=t;
        graf[j].push_back(k);
        graf[k].push_back(j);
    }

    max_flow=0;
    while(nr=dfs(1,INF))
    {
        max_flow+=nr;
        while(!noduri_vizitate_in_parcurgere.empty())
        {
            vizitat[noduri_vizitate_in_parcurgere.front()]=0;
            noduri_vizitate_in_parcurgere.pop();
        }
    }


    fout<<max_flow<<'\n';



    return 0;
}