Cod sursa(job #405867)

Utilizator Alexa_ioana_14Antoche Ioana Alexandra Alexa_ioana_14 Data 28 februarie 2010 20:38:48
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.75 kb
/*#include <stdio.h>
#include <vector>

using namespace std;

#define NMAX 1024
#define pb push_back
#define sz size()
#define mp make_pair
#define INF 0x3f3f3f3f

int C[NMAX][NMAX];
int F[NMAX][NMAX];
int TT[NMAX];
vector<int> G[NMAX];
int N, M;
int cd[NMAX];
int viz[NMAX];

int BF()
{
	int i, j, nod, V;

	cd[0] = 1;
	cd[1] = 1;
	memset(viz, 0, sizeof(viz));
	viz[1] = 1;

	for (i = 1; i <= cd[0]; i++)
	{
		nod = cd[i];
                if (nod == N) continue;
		for (j = 0; j < G[nod].sz; j++) 
			{
				V = G[nod][j];
				if (C[nod][V] == F[nod][V] || viz[V]) continue;
				viz[V] = 1;
				cd[ ++cd[0] ] = V;
				TT[V] = nod;
			}
	}

	return viz[N];
}

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

	int i, flow, fmin, x, y, z, nod;

	scanf("%d %d ", &N, &M);

	for (i = 1; i <= M; i++)
	{
		scanf("%d %d %d ", &x, &y, &z);
		C[x][y] += z;
		G[x].pb(y);
		G[y].pb(x);
	}

	for (flow = 0; BF();)
		for (i = 0; i < G[N].sz; i++)
		{
			nod = G[N][i];
			if (F[nod][N] == C[nod][N] || !viz[nod]) continue;
			TT[N] = nod;
			
			fmin = INF;
			for (nod = N; nod != 1; nod = TT[nod]) 
				fmin = min(fmin, C[ TT[nod] ][nod] - F[ TT[nod] ][nod]);
			if (fmin == 0) continue;
			
			for (nod = N; nod != 1; nod = TT[nod])
			{
				F[ TT[nod] ][nod] += fmin;
				F[nod][ TT[nod] ] -= fmin;
			}

			flow += fmin;
		}

	printf("%d ", flow);
	return 0;
}*/
#include<cstdio>
#include<vector>
#include<bitset>
using namespace std;
#define NMAX 1005
#define minn(a,b) ((a<b)?a:b)
#define OO (1<<30)
#define sh short int
#define pb push_back
vector<sh>a[NMAX];
bitset<NMAX>viz;
sh N,x,y,nod,t[NMAX],q[NMAX];
int M,capacitate,r,c[NMAX][NMAX],f[NMAX][NMAX];
bool bf()
{
	int p=0,u=0;
	q[u++]=1;
	viz.reset();
	viz[1]=1;
	while (u!=p)
	{
		nod=q[p++];
		if (nod==N)
			continue;
		size_t g=a[nod].size();
		for (size_t i=0; i<g; ++i)
		{
			sh fiu=a[nod][i];
			if (c[nod][fiu]==f[nod][fiu] || viz[fiu])
				continue;
			q[u++]=fiu;
			viz[fiu]=1;
			t[fiu]=nod;
		}
	}
	return viz[N]==1;
}
int main()
{
	freopen("maxflow.in","r",stdin);
	freopen("maxflow.out","w",stdout);
	scanf("%hd%d",&N,&M);
	while (M--)
	{
		scanf("%hd%hd%d",&x,&y,&capacitate);
		c[x][y]+=capacitate;
		a[x].pb(y);
		a[y].pb(x);
	}
	M=0;
	size_t g=a[N].size();
	while(bf())
	{
		for (size_t i=0; i<g; ++i)
		{
			nod=a[N][i];
			if (c[nod][N]==f[nod][N] || !viz[nod])
				continue;
			t[N]=nod;
			r=OO;
			for (nod=N; nod!=1; nod=t[nod])
				r=minn(r,c[t[nod]][nod]-f[t[nod]][nod]);
			if (!r)
				continue;
			for (nod=N; nod!=1; nod=t[nod])
			{
				f[t[nod]][nod]+=r;
				f[nod][t[nod]]-=r;
			}
			M+=r;
		}
	}
	printf("%d",M);
	return 0;
}