Cod sursa(job #1659822)

Utilizator zertixMaradin Octavian zertix Data 22 martie 2016 17:25:21
Problema Flux maxim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.52 kb
#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;

vector <int> g[1002];
int n,m,tata[1002],flux[1002][1002],capacitate[1002][1002],total;
queue < int > q;
void citire()
{
    scanf("%d%d",&n,&m);
    int nod1,nod2,cost;
    for (int i=1;i<=m;++i)
    {
        scanf("%d%d%d",&nod1,&nod2,&cost);
        g[nod1].push_back(nod2);
        g[nod2].push_back(nod1);
        capacitate[nod1][nod2]=cost;
    }
}
int bfs()
{
    q.push(1);
    memset(tata,0,sizeof(tata));
    tata[1]=1;
    while (!q.empty())
    {
        int nod=q.front();
        q.pop();

        if (nod==n)
            continue;
        for (vector <int>:: iterator it=g[nod].begin();it!=g[nod].end();++it)
            {
                if (tata[*it] || capacitate[nod][*it]==flux[nod][*it])
                    continue;
                tata[*it]=nod;
                q.push(*it);
            }
    }
    return tata[n];
}
int main()
{
    freopen("maxflow.in","r",stdin);
    freopen("maxflow.out","w",stdout);
    citire();
    while (1)
    {
        int l=bfs();
        if (l==0)
            break;
        l=n;
        int minim=inf;
        while (l!=1)
        {
            minim=min(minim,capacitate[tata[l]][l]-flux[tata[l]][l]);
            l=tata[l];
        }
        l=n;
        while (l!=1)
        {
            flux[tata[l]][l]+=minim;
            flux[l][tata[l]]-=minim;
            l=tata[l];
        }
        total+=minim;
    }
    printf("%d",total);
    return 0;
}