Cod sursa(job #1161731)

Utilizator lianaliana tucar liana Data 31 martie 2014 13:47:46
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.95 kb
#include<stdio.h>
#include<vector>
using namespace std;
#define nmax 1005
#define inf 210000
int n, m, x, y, z, i, nbf, inc, sf, fmin, rez, ntn, el, nod;
vector <int> ma[nmax];
vector <int> ::iterator it;
int co[nmax], t[nmax], viz[nmax], tn[nmax], cap[nmax][nmax];

void citire()
{
    scanf("%ld %ld",&n,&m);
    for (i=1;i<=m;i++)
    {
        scanf("%ld %ld %ld",&x,&y,&z);
        ma[x].push_back(y); cap[x][y]=z;
        ma[y].push_back(x);
    }
}

void bfs()
{
    nbf++;  ntn=0;
    co[1]=sf=inc=1;
    viz[1]=nbf;
    while (inc<=sf)
    {
        el=co[inc]; inc++;
        if (el==n)
            break;
        for (it=ma[el].begin();it!=ma[el].end();it++)
            if (cap[el][*it]>0)
            {
                if (viz[*it]!=nbf)
                {
                    co[++sf]=*it;
                    viz[*it]=nbf;
                    t[*it]=el;
                }
                if ((*it)==n)
                    tn[++ntn]=el;
            }
    }
}

void rezolvare()
{
    while (1)
    {
        bfs();
        if (viz[n]==nbf)
            for (i=1;i<=ntn;i++)
            {
                t[n]=tn[i];
                nod=n;  fmin=inf;
                while ((nod>1)&&(fmin>0))
                {
                    if (fmin>cap[t[nod]][nod])
                        fmin=cap[t[nod]][nod];
                    nod=t[nod];
                }
                if (fmin>0)
                {
                    nod=n;
                    while (nod>1)
                    {
                        cap[t[nod]][nod]-=fmin;
                        cap[nod][t[nod]]+=fmin;
                        nod=t[nod];
                    }
                }
                rez+=fmin;
            }
        else
            break;
    }
}

int main()
{
    freopen("maxflow.in","r",stdin);
    freopen("maxflow.out","w",stdout);
    citire();
    rezolvare();
    printf("%ld",rez);
    return 0;
}