Cod sursa(job #357435)

Utilizator DraStiKDragos Oprica DraStiK Data 19 octombrie 2009 11:53:48
Problema Critice Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.27 kb
#include <stdio.h>

#define DIM 1005
#define oo 1<<30-1

struct nod {int x;
            nod *urm;} *lst[DIM];
int c[DIM][DIM],f[DIM][DIM];
int q[DIM],t[DIM],viz[DIM],sol[10*DIM];
int n,m,nrt;

void add (int a,int b)
{
    nod *p=new nod;
    p->x=a;
    p->urm=lst[b];
    lst[b]=p;
}

void read ()
{
	freopen ("critice.in","r",stdin);
	int i,x,y,z;

	scanf ("%d%d",&n,&m);
	for (i=1; i<=m; ++i)
	{
		scanf ("%d%d%d",&x,&y,&z);
		c[x][y]=c[y][x]=z;
		add (x,y);
		add (y,x);
	}
}

inline void init ()
{
    int i;

    for (i=1; i<=n; ++i)
        t[i]=0;
}

inline int bf ()
{
    int st,dr,i;
    nod *p;

    init ();
    for (q[st=dr=1]=1; st<=dr; ++st)
        for (p=lst[q[st]]; p; p=p->urm)
            if (!t[p->x] && c[q[st]][p->x]>f[q[st]][p->x])
            {
                t[p->x]=q[st];
                q[++dr]=p->x;
            }
    return t[n];
}

int min (int a,int b)
{
    if (a<b)
        return a;
    return b;
}

void flux ()
{
    int i,j,minim;

    for ( ; bf (); )
		for (i=1; i<=n; ++i)
		    if (t[i]>0 && c[i][n]-f[i][n]>0)
		    {
				minim=c[i][n]-f[i][n];
    			for (j=i; j!=1; j=t[j])
    				minim=min (minim,c[t[j]][j]-f[t[j]][j]);
                f[i][n]+=minim;
                f[n][i]-=minim;
    			for (j=i; j!=1; j=t[j])
                {
                    f[t[j]][j]+=minim;
                    f[j][t[j]]-=minim;
                }
        	}
}

inline int abs (int a)
{
    if (a<0)
        return -a;
    return a;
}

void df (int start,int tip)
{
    nod *p;

    t[start]=1;
    viz[start]|=tip;
    for (p=lst[start]; p; p=p->urm)
        if (!t[p->x] && c[start][p->x]>abs (f[start][p->x]))
            df (p->x,tip);
}

void solve ()
{
    freopen ("critice.in","r",stdin);
    freopen ("critice.out","w",stdout);
    int i,x,y,z;

    flux ();
    init ();
    df (1,1);
    init ();
    df (n,2);
    scanf ("%d%d",&n,&m);
	for (i=1; i<=m; ++i)
	{
		scanf ("%d%d%d",&x,&y,&z);
        if ((viz[x]|viz[y])==3)
            sol[++nrt]=i;
	}
}

void print ()
{
    int i;

    printf ("%d\n",nrt);
    for (i=1; i<=nrt; ++i)
        printf ("%d\n",sol[i]);
}

int main ()
{
    read ();
    solve ();
    print ();

    return 0;
}