Cod sursa(job #2671130)

Utilizator bem.andreiIceman bem.andrei Data 11 noiembrie 2020 15:12:36
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.65 kb
#include <bits/stdc++.h>
#define NMAX 1000
#define MAXVAL (1LL<<31)-1;

using namespace std;
ifstream r("maxflow.in");
ofstream w("maxflow.out");

int n,m,x,y,z;
int sol,minf;
int capacity[NMAX+5][NMAX+5],flow[NMAX+5][NMAX+5];
int viz[NMAX+5],p[NMAX+5];
vector<int> g[NMAX+5];
queue<int> q;
bool bfs(){
    while(!q.empty()){
        q.pop();
    }
    for(int i=1;i<=n;i++){
        viz[i]=0;
    }
    q.push(1);
    viz[1]=1;
    int node,node2;
    while(!q.empty()){
        node=q.front();
        q.pop();
        if(node!=n){
            for(int i=0;i<g[node].size();i++){
                node2=g[node][i];
                if(capacity[node][node2]-flow[node][node2]>0&&!viz[node2]){
                    viz[node2]=1;
                    p[node2]=node;
                    q.push(node2);
                }
            }
        }
    }
    return viz[n];
}
int main(){
    r>>n>>m;
    for(int i=1;i<=m;i++){
        r>>x>>y>>z;
        g[x].push_back(y);
        g[y].push_back(x);
        capacity[x][y]+=z;
    }
    while(bfs()){
        for(int j=0;j<g[n].size();j++){
            if(capacity[g[n][j]][n]-flow[g[n][j]][n]>0&&viz[g[n][j]]){
                p[n]=g[n][j];
                minf=MAXVAL;
                for(int i=n;i!=1;i=p[i]){
                    minf=min(minf,capacity[p[i]][i]-flow[p[i]][i]);
                }
                if(minf){
                    for(int i=n;i!=1;i=p[i]){
                        flow[p[i]][i]+=minf;
                        flow[i][p[i]]-=minf;
                    }
                    sol+=minf;
                }
            }
        }
    }
    w<<sol<<"\n";
}