Cod sursa(job #252764)

Utilizator yoyolichIoana Ardeleanu yoyolich Data 4 februarie 2009 21:36:28
Problema Flux maxim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include<stdio.h>
#include<vector>
#define nmax 1024
#define oo 0x3f3f3f3f
using namespace std;

FILE *f=fopen("maxflow.in","r"), *g=fopen("maxflow.out","w");

int n,m,C[nmax][nmax],viz[nmax],flow,fmin,x,y,z,tata[nmax],F[nmax][nmax],i;
int Q[nmax];
vector<int> G[nmax];
/*inline int min(int x, int y)
{
	if(x<y) return x;
	return y;
}*/

void citire()
{
	fscanf(f,"%d %d",&n,&m);
	for(int i=1;i<=m;i++)
	{
		fscanf(f,"%d %d %d",&x,&y,&z);
		G[x].push_back(y);
		G[y].push_back(x);
		C[x][y]+=z;
	}
	fclose(f);
}

inline int bfs()
{
	int prim, ultim,x,i;

	memset(viz,0,sizeof(viz));
	prim=ultim=0;
	Q[0]=1;
	viz[1]=1;
	while(prim<=ultim)
	{
		x=Q[prim++];
		if(x==n) continue;
		for(i=0;i<G[x].size();i++)
				{
					if (C[x][G[x][i]] == F[x][G[x][i]] || viz[G[x][i]]) continue;  
					Q[++ultim]=G[x][i];
					tata[G[x][i]]=x;
					viz[G[x][i]]=1;
				}
	}
	return viz[n];
}

int main()
{
	citire();

	
	flow=0;
	while(bfs())
	   for(i=0;i<G[n].size();i++)
		{
		if (F[G[n][i]][n] == C[G[n][i]][n] || !viz[G[n][i]]) continue;
		tata[n]=G[n][i];
		
		fmin=oo;
		for(i=n;i!=1;i=tata[i])
		{
			fmin=min(fmin, C[tata[i]][i]-F[tata[i]][i]);
		}
		if (fmin == 0) continue;
		
		for(i=n;i!=1;i=tata[i])
		{
			F[tata[i]][i]+=fmin;
			F[i][tata[i]]-=fmin;
		}
		
		flow+=fmin;
	}

	
	fprintf(g,"%d\n",flow);
	fclose(g);
	return 0;
}