Cod sursa(job #39983)

Utilizator vlad_popaVlad Popa vlad_popa Data 27 martie 2007 10:22:10
Problema Critice Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.27 kb
using namespace std;

#include <cstdio>
#include <cassert>
#include <string>

#define FIN "critice.in"
#define FOUT "critice.out"
#define NMAX 1001
#define MMAX 10001
#define INF 0x3f3f3f3f

int c[NMAX][NMAX], N, M, f[NMAX][NMAX], vizs[NMAX], vizt[NMAX];
int pred[NMAX], rez[MMAX], cd[NMAX], begin, ended;
int retn1[MMAX], retn2[MMAX];
struct nod {int vf;
            nod *next;};
typedef nod *pnod;
pnod a[NMAX];

void add (int x, int y)
{
  pnod p = new nod;
  p -> vf = y;
  p -> next = a[x];
  a[x] = p;
}

int bf ()
{
  int v, i;
  pnod p;
  
  memset (pred, 0, sizeof (pred));
  
  pred[1] = -1;
  begin  = ended = 0;
  cd[ended++] = 1;
  
  while (begin != ended)
   {
     v = cd[begin++];
     assert (v >= 1 && v <= N);
     for (p = a[v]; p; p = p -> next)
      {
        i = p -> vf;
        if (c[v][i] - f[v][i] > 0 && !pred[i])
         {
           pred[i] = v;
           cd[ended++] = i;
           
           if (i == N)
            return 1;
         }
      }
   }
  return 0;
}

int MIN (int x, int y)
{
  return x < y ? x : y;
}

void ford_fulk ()
{
  int i, min;
  
  while (bf ())
   {
     min = INF;
     
     for (i = N; pred[i] != -1; i = pred[i])
       min = MIN (min, c[pred[i]][i] - f[pred[i]][i]);
         
     for (i = N; pred[i] != -1; i = pred[i])
      {
        f[pred[i]][i] += min;
        f[i][pred[i]] -= min;
      }
   }
}

void bfs ()
{
  int v, i;
  pnod p;

  memset (pred, 0, sizeof (pred));
  pred[1] = -1;
  vizs[1] = true;
  begin = ended = 0;
  cd[ended++] = 1;
  
  while (begin != ended)
   {
     v = cd[begin++];
     //fprintf (stderr, "%d ", v);
     assert (v >= 1 && v <= N);
     for (p = a[v]; p; p = p -> next)
      {
        i = p -> vf;
        if (c[v][i] - f[v][i] > 0 && -f[v][i] != c[v][i] && !pred[i])
         {
           pred[i] = v;
           vizs[i] = true;
           cd[ended++] = i;
         }
      }
   }
}

void bft ()
{
  int v, i;
  pnod p;

  memset (pred, 0, sizeof (pred));
  
  pred[N] = -1;
  vizt[N] = true;
  begin = ended = 0;
  cd[ended++] = N;
  
  while (begin != ended)
   {
     v = cd[begin++];
     //fprintf (stderr, "%d ", v);
     assert (v >= 1 && v <= N);
     for (p = a[v]; p; p = p -> next)
      {
        i = p -> vf;
        if (c[v][i] - f[v][i] > 0 && -f[v][i] != c[v][i] && !pred[i])
         {
           pred[i] = v;
           vizt[i] = true;
           cd[ended++] = i;
         }
      }
   }
}

int
 main ()
{
  int x, y, cap;
  freopen (FIN, "rt", stdin);
  freopen (FOUT, "wt", stdout);

  scanf ("%d%d", &N, &M);
  for (int i = 1; i <= M; ++ i)
   {
     scanf ("%d%d%d", &x, &y, &cap);
     c[x][y] = c[y][x] = cap;
     add (x, y);
     add (y, x);
     retn1[i] = x;
     retn2[i] = y;
   }
  ford_fulk ();
  /*for (int i = 1; i <= N; ++ i)
    for (int j = 1; j <= N; ++ j)
      if (f[i][j] > 0)
        printf ("%d %d    %d\n", i, j, f[i][j]);*/
  bfs ();
  bft ();
  int ct = 0;
  for (int i = 1; i <= M; ++ i)
    if ((vizs[retn1[i]] && vizt[retn2[i]]) ||     
         (vizs[retn2[i]] && vizt[retn1[i]]))
           rez[++ct] = i;
  printf ("%d\n", ct);
  for (int i = 1; i <= ct; ++ i)
    printf ("%d\n", rez[i]);
  return 0;
}