Cod sursa(job #558567)

Utilizator acelasi7Tudor Maxim acelasi7 Data 17 martie 2011 12:47:01
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.6 kb
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
#define nrn 1005
#define inf INT_MAX

vector<int>V[nrn];
queue<int>Q;
int C[nrn][nrn],F[nrn][nrn],grint[nrn],grext[nrn],flux,n;
int tata[nrn],uz[nrn];

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

int bfs()
{
	int i,node,v;
	memset(uz,0,sizeof(uz));
	/*for(i=0;i<V[0].size();++i)
	{
		Q.push(V[0][i]);
		tata[V[0][i]]=V[0][i];
		uz[V[0][i]]=1;
	}*/
	Q.push(1);
	tata[1]=1;
	uz[1]=1;
	while(!Q.empty())
	{
		node=Q.front();
		Q.pop();
		if(node==n)
			continue;
		for(i=0;i<V[node].size();++i)
		{
			v=V[node][i];
			if(!(C[node][v]==F[node][v])&&!uz[v])
			{
				uz[v]=1;
				tata[v]=node;
				Q.push(v);
			}
		}
	}
	return uz[n];
}
int main()
{
	int a,b,cost,m,nod,tt,i;
	fscanf(in,"%d%d",&n,&m);
	for(i=1;i<=m;++i)
	{
		fscanf(in,"%d%d%d",&a,&b,&cost);
		V[a].push_back(b);
		V[b].push_back(a);
		C[a][b]=cost;
		grext[a]++;
		grint[b]++;
	}
	/*for(i=1;i<=n;++i)
		if(!grint[i])
			V[0].push_back(i);*/
	while(bfs())
	{
		nod=n;
		for(i=0;i<V[n].size();++i)
		{
			nod=V[n][i];
			if(!uz[nod]||(F[nod][n]==C[nod][n]))
				continue;
			tata[n]=nod;
			nod=n;
			tt=tata[nod];
			cost=inf;
			while(tt!=nod)
			{
				cost=min(cost,C[tt][nod]-F[tt][nod]);
				nod=tt;
				tt=tata[tt];
			}
			flux+=cost;
			nod=n;
			tt=tata[nod];
			if(!cost) continue;
			F[tt][nod]+=cost;
			F[nod][tt]-=cost;
			while(tt!=nod)
			{
				nod=tt;
				tt=tata[tt];
				F[tt][nod]+=cost;
				F[nod][tt]-=cost;
			}
		}
	}
	fprintf(out,"%d\n",flux);
	return 0;
}