Cod sursa(job #1996062)

Utilizator AlexandruRudiAlexandru Rudi AlexandruRudi Data 29 iunie 2017 20:14:53
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.61 kb
#include <bits/stdc++.h>
using namespace std;
#define x first
#define y second
#define dbg(x) cout << #x << '=' << x << '\n';
#define ll long long
#define pi pair<int,int>
#define pl pair<long long,long long>
#define rd(x) cin >> x;
#define rda(a,n) for(int i=1;i<=n;i++) cin >> a[i];
#define wr(x) cout << x << ' ';
#define wrl(x) cout << x << '\n';
#define wra(a,n) for(int i=1;i<=n;i++) cout << a[i] << ' '; cout << '\n';
#define lg length()
#define pb push_back
ifstream in("maxflow.in");
ofstream out("maxflow.out");
#define MAXN 100005
#define INF 1000000005
#define LINF 1000000000000000005

int n,m,c[1005][1005],f[1005][1005],v[1005],x,y,z,ans,p[1005];

vector <int> g[1005];
queue <int> l;

bool BFS(){
    l.push(1); v[1]=1;
    while(l.size()){
        int nod=l.front();
        l.pop();
        if(nod==n) continue;
        for(int i : g[nod]){
            if(c[nod][i]==f[nod][i] || v[i]) continue;
            l.push(i); p[i]=nod; v[i]=1;
        }
    }
    return v[n];
}

int32_t main(){
    ios_base :: sync_with_stdio(0);
    in >> n >> m;
    for(int i=1;i<=m;i++){
        in >> x >> y >> z;
        g[x].pb(y);
        g[y].pb(x);
        c[x][y]=z;
    }
    while(BFS()){
        for(int i : g[n]){
            if(c[i][n]==f[i][n] || !v[i]) continue;
            int fl=1e9;
            p[n]=i;
            for(int j=n;j!=1;j=p[j]) fl=min(fl,c[p[j]][j]-f[p[j]][j]);
            if(fl==0) continue;
            for(int j=n;j!=1;j=p[j]) f[p[j]][j]+=fl,f[j][p[j]]-=fl;
            ans+=fl;
        }
        for(int i=1;i<=n;i++) v[i]=0;
    }
    out << ans;
}