Cod sursa(job #2124432)

Utilizator alexilasiAlex Ilasi alexilasi Data 7 februarie 2018 11:01:22
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.27 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("maxflow.in");
ofstream fout("maxflow.out");

int n,m,A,B,C,i,ans,fc,j;

int t[1001];

int c[1001][1001],f[1001][1001];

vector <int> v[1001];

bool bf()
{
    int nod;
    queue <int> q;
    memset(t,0,sizeof(t));
    t[1]=-1;q.push(1);
    while(!q.empty())
    {
        nod=q.front();q.pop();
        for(auto it:v[nod])
        {
            if(t[it]||c[nod][it]==f[nod][it])continue;
            t[it]=nod;
            q.push(it);
            if(it==n)return 1;
        }
    }
    return 0;
}

int main()
{
    fin>>n>>m;
    for(i=1;i<=m;i++)
    {
        fin>>A>>B>>C;
        v[A].push_back(B);
        v[B].push_back(A);
        c[A][B]=C;
    }
    while(bf())
    {
        for(auto it:v[n])
        {
            if(!t[it]||c[it][n]==f[it][n]||!c[it][n])continue;
            fc=c[it][n]-f[it][n];
            for(j=it;j>1;j=t[j])
                fc=min(fc,c[t[j]][j]-f[t[j]][j]);
            if(!fc)continue;
            f[it][n]+=fc;
            f[n][it]-=fc;
            for(j=it;j>1;j=t[j])
            {
                f[t[j]][j]+=fc;
                f[j][t[j]]-=fc;
            }
            ans+=fc;
        }
    }
    fout<<ans;
    return 0;
}