Cod sursa(job #2380592)

Utilizator SchnitzelMannPavaloiu Gabriel SchnitzelMann Data 15 martie 2019 10:57:43
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.37 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in("maxflow.in");
ofstream out("maxflow.out");
int viz[1001],pr[1001],f[1001][1001],c[1001][1001],n;
vector<int> v[1001];
queue<int> q;
bool bfs()
{
    int x;
    memset(viz,0,sizeof(viz));
    memset(pr,0,sizeof(pr));
    q.push(1);
    viz[1]=1;
    while(!q.empty())
    {
        x=q.front();
        q.pop();
        if(x==n)
            break;
        for(auto it:v[x])
            if(!viz[it]&&f[x][it]<c[x][it])
            {
                viz[it]=1;
                pr[it]=x;
                q.push(it);
            }
    }
    while(!q.empty())
        q.pop();
    return viz[n];
}
int main()
{
    int m,nr1,nr2,nr;
    in>>n>>m;
    while(m--)
    {
        in>>nr1>>nr2>>nr;
        v[nr1].push_back(nr2);
        v[nr2].push_back(nr1);
        c[nr1][nr2]=nr;
    }
    nr=0;
    while(bfs())
        for(auto it:v[n])
            if(viz[it]&&f[it][n]!=c[it][n])
            {
                pr[n]=it;
                nr1=1<<29;
                for(nr2=n;nr2>1;nr2=pr[nr2])
                    nr1=min(nr1,c[pr[nr2]][nr2]-f[pr[nr2]][nr2]);
                for(nr2=n;nr2>1;nr2=pr[nr2])
                {
                    f[pr[nr2]][nr2]+=nr1;
                    f[nr2][pr[nr2]]-=nr1;
                }
                nr+=nr1;
            }
    out<<nr;
    return 0;
}