Cod sursa(job #357415)

Utilizator DraStiKDragos Oprica DraStiK Data 19 octombrie 2009 11:08:20
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.36 kb
#include <stdio.h>

#define DIM 1005
#define oo 11005

int c[DIM][DIM],f[DIM][DIM];
int q[DIM],t[DIM];
int n,m,minim,flow;

void read ()
{
	int i,x,y,z;

	scanf ("%d%d",&n,&m);
	for (i=1; i<=m; ++i)
	{
		scanf ("%d%d%d",&x,&y,&z);
		c[x][y]=z;
	}
}

int bf ()
{
    int st,dr,i;

    for (i=1; i<=n; ++i)
        t[i]=0;
    for (q[st=dr=1]=1; st<=dr; ++st)
        for (i=1; i<=n; ++i)
            if (!t[i] && c[q[st]][i]>f[q[st]][i])
            {
                    t[i]=q[st];
                    q[++dr]=i;
                if (i==n)
                    return 1;
            }
    return 0;
}

int min (int a,int b)
{
    if (a<b)
        return a;
    return b;
}

void solve ()
{
    int i,j;

    for ( ; bf (); )
		for (i=1; i<=n; ++i)
		    if (t[i]>0 && c[i][n]-f[i][n]>0)
		    {
				minim=c[i][n]-f[i][n];
    			for (j=i; j!=1; j=t[j])
    				minim=min (minim,c[t[j]][j]-f[t[j]][j]);
                f[i][n]+=minim;
                f[n][i]-=minim;
    			for (j=i; j!=1; j=t[j])
                {
                    f[t[j]][j]+=minim;
                    f[j][t[j]]-=minim;
                }
                flow+=minim;
        	}

    printf ("%d",flow);
}

int main ()
{
    freopen ("maxflow.in","r",stdin);
    freopen ("maxflow.out","w",stdout);

    read ();
    solve ();

    return 0;
}