Cod sursa(job #92508)

Utilizator MariusMarius Stroe Marius Data 15 octombrie 2007 19:37:13
Problema Critice Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.65 kb
#include <stdio.h>

const char iname[] = "critice.in";
const char oname[] = "critice.out";

#define MAX_N 1001
#define MAX_M 10000

int n, *G[MAX_N], m, E[MAX_M][2], deg[MAX_N], C[MAX_N][MAX_N], F[MAX_N][MAX_N];

int Q[MAX_N], T[MAX_N];


void read_in(void)
{
	int x, y, c;

	FILE *fi = fopen(iname, "r");

	fscanf(fi, "%d", &n);
	fscanf(fi, "%d", &m);
	for (int i = 0; i < m; ++ i) 
	{
		fscanf(fi, "%d %d %d", &x, &y, &c);
		C[x][y] = C[y][x] = c;
		E[i][0] = x;
		E[i][1] = y;
		deg[x] ++, deg[y] ++;
	}
	for (int i = 1; i <= n; ++ i) 
	{
		G[i] = new int [deg[i] + 1];
		G[i][deg[i]] = -1;
		deg[i] = 0;
	}
	for (int i = 0; i < m; ++ i) 
	{
		x = E[i][0];
		y = E[i][1];
		G[x][deg[x] ++] = y;
		G[y][deg[y] ++] = x;
	}
	fclose(fi);
}

int BFS(void)
{
	int head, tail;

	for (int i = 2; i <= n; ++ i) 
		T[i] = -1;
	for (Q[head = tail = 0] = 1; head <= tail; ++ head) 
	{
		int x = Q[head];
		for (int *p = G[x]; *p != -1; ++ p) 
		{
			if (T[*p] == -1 && C[x][*p] - F[x][*p] > 0)
			{
				T[*p] = x;
				Q[++ tail] = *p;
				if (*p == n)
					return 1;
			}
		}
	}
	return 0;
}

void Dinic(void)
{
	while (BFS())
	{
		for (int i = 1; i <= n; ++ i)
		{
			if (T[i] == -1 || C[i][n] <= F[i][n])
				continue ;
			int r = C[i][n] - F[i][n];
			for (int j = i; j != 1; j = T[j])
			{
				if (r > C[T[j]][j] - F[T[j]][j])
					r = C[T[j]][j] - F[T[j]][j];
			}
			if (r == 0)
				continue ;
			F[i][n] += r, F[n][i] -= r;
			for (int j = i; j != 1; j = T[j]) 
			{
				F[T[j]][j] += r;
				F[j][T[j]] -= r;
			}
		}
	}
}

void BF1(int start, int val)
{
	int head, tail;

	T[start] = val;
	for (Q[head = tail = 0] = start; head <= tail; ++ head)
	{
		int x = Q[head];
		for (int *p = G[x]; *p != -1; ++ p)
		{
			if (T[*p] != val && C[x][*p] - F[x][*p] > 0)
			{
				Q[++ tail] = *p;
				T[*p] = val;
			}
		}
	}
}

void BF2(int start, int val)
{
	int head, tail;

	T[start] = val;
	for (Q[head = tail = 0] = start; head <= tail; ++ head)
	{
		int x = Q[head];
		for (int *p = G[x]; *p != -1; ++ p)
		{
			if (T[*p] != val && C[*p][x] - F[*p][x] > 0)
			{
				Q[++ tail] = *p;
				T[*p] = 2;
			}
		}
	}
}

int main(void)
{
	read_in();
	Dinic();

	FILE *fo = fopen(oname, "w");

	for (int i = 1; i <= n; ++ i)
		T[i] = -1;
	BF1(1, 1);
	BF2(n, 2);
	int res = 0;
	for (int i = 0; i < m; ++ i) 
	{
		int x = E[i][0];
		int y = E[i][1];
		if (T[x] + T[y] == 3)
			res ++;
	}
	fprintf(fo, "%d\n", res);
	for (int i = 0; i < m; ++ i) 
	{
		int x = E[i][0];
		int y = E[i][1];
		if (T[x] + T[y] == 3)
			fprintf(fo, "%d\n", i + 1);
	}
	fclose(fo);

	return 0;
}