Cod sursa(job #528927)

Utilizator PlayLikeNeverB4George Marcus PlayLikeNeverB4 Data 3 februarie 2011 21:00:26
Problema Critice Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.59 kb
#include <stdio.h>
#define maxn 1024
#define maxm 10005
#define oo 1000000

struct muchii {
	int a,b;
} id[maxm];

int i,N,M;
int A[maxn][maxn],C[maxn][maxn],from[maxn],x[maxn],y[maxn],sol[maxm];

void citire()
{
	int x,y,c;
	scanf("%d %d",&N,&M);
	for(i=1;i<=M;i++)
	{
		scanf("%d %d %d",&x,&y,&c);
		A[x][++A[x][0]]=y;
		A[y][++A[y][0]]=x;
		C[y][x]=C[x][y]=c;
		id[i].a=x; id[i].b=y;
	}
}

inline int min(int a, int b)
{
	return (a>b)? b : a;
}

int bfs()
{
	int pi,ps,k,cd[maxn]; ps=pi=1;
	for(i=1;i<=N;i++) from[i]=0;
	cd[1]=1;
	while(ps<=pi)
	{
		k=cd[ps];
		for(i=1;i<=A[k][0];i++)
			if(from[A[k][i]]==0 && C[k][A[k][i]]>0)
			{
				cd[++pi]=A[k][i];
				from[A[k][i]]=k;
				if(A[k][i]==N) return 1;
			}
		ps++;
	}
	return 0;
}

void search(int st,int v[])
{
	int pi,ps,k,cd[maxn];
	for(i=1;i<=N;i++) v[i]=0;
	pi=ps=1; cd[1]=st; v[st]=1;
	while(ps<=pi)
	{
		k=cd[ps];
		for(i=1;i<=A[k][0];i++)
			if(v[A[k][i]]==0 && C[k][A[k][i]]>0 && C[A[k][i]][k])
			{
				cd[++pi]=A[k][i];
				v[A[k][i]]=1;
			}
		ps++;
	}
}

int main()
{
	freopen("critice.in","r",stdin);
	freopen("critice.out","w",stdout);
	citire();
	while(bfs())
	{
		int k=N,flux=oo;
		do
		{
			flux=min(flux,C[from[k]][k]);
			k=from[k];
		}
		while(k!=1);
		k=N;
		do
		{
			C[from[k]][k]-=flux;
			C[k][from[k]]+=flux;
			k=from[k];
		}
		while(k!=1);
	}
	search(1,x);
	search(N,y);
	for(i=1;i<=M;i++)
		if(x[id[i].a] && y[id[i].b] || x[id[i].b] && y[id[i].a])
			sol[++sol[0]]=i;
	printf("%d\n",sol[0]);
	for(i=1;i<=sol[0];i++)
		printf("%d\n",sol[i]);
}