Cod sursa(job #355794)

Utilizator DraStiKDragos Oprica DraStiK Data 12 octombrie 2009 10:53:36
Problema Flux maxim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.48 kb
#include <stdio.h>

#define DIM 1005
#define oo 11005

struct nod {int x;
            nod *urm;} *lst[DIM];
int c[DIM][DIM],f[DIM][DIM];
int q[DIM],viz[DIM],t[DIM];
int n,m,minim,flow;

void add (int a,int b)
{
    nod *p=new nod;
    
    p->x=a;
    p->urm=lst[b];
    lst[b]=p;
}

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;
		add (x,y);
		add (y,x);
	}
}

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

void init ()
{
    int i;
    
    for (i=1; i<=n; ++i)
        viz[i]=0;
    minim=oo;
}

int bf ()
{
    nod *p;
    int st,dr;
    
    for (q[st=dr=1]=1; st<=dr; ++st)
        for (p=lst[q[st]]; p; p=p->urm)
            if (c[q[st]][p->x]!=f[q[st]][p->x] && !viz[p->x])
            {
                viz[p->x]=1;
                q[++dr]=p->x;
                t[p->x]=q[st];
                if (p->x==n)
                    return 1;
            }
    return 0;
}

void solve ()
{
    int i;
    
    for (minim=oo; bf (); flow+=minim, init ())
    {
        for (i=n; i!=1; i=t[i])
            minim=min (minim,c[t[i]][i]-f[t[i]][i]);
        for (i=n; i!=1; i=t[i])
        {
            f[t[i]][i]+=minim;
            f[i][t[i]]-=minim;
        }
    }
    printf ("%d",flow);
}

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

    read ();
    solve ();

    return 0;
}