Cod sursa(job #245286)

Utilizator runnaway90Oprescu Radu Constantin runnaway90 Data 17 ianuarie 2009 15:49:20
Problema Flux maxim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.6 kb
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define N 1024
#define INF 100000000

long q[N], c[N][N], tata[N], n, nr, sol[N],*a[N];

int bfs(){
long x,p,u,i,j;
     memset(tata,0,N * sizeof(long));
     q[0]=1;tata[1]=1;nr=0;
     for (p=0,u=0;p<=u;p++){
         x=q[p];
         for (j=1;j<=a[x][0];j++)
             if (c[x][a[x][j]]>0 &&!tata[a[x][j]]){
                tata[a[x][j]]=x;q[++u]=a[x][j];
                if (tata[n]!=0){ u--,tata[n]==0; sol[++nr]=x;} 
             }     
     }
     return (tata[n]);
}

int main (){
long x,y,z,flux=0,i,m,min,w;
    freopen("maxflow.in","r",stdin);
    freopen("maxflow.out","w",stdout);
    
    scanf("%d %d ",&n,&m);
    
    for (i=1;i<=n;i++){a[i]=(long*) malloc(sizeof(long));a[i][0]=0;}
    
    for (i=1;i<=m;i++){ scanf("%d %d %d",&x,&y,&z);
        c[x][y]=z;
        
        a[x][0]++;
        a[x]=(long*) realloc (a[x],(a[x][0]+1)*sizeof(long));
        a[x][a[x][0]]=y;

        a[y][0]++;
        a[y]=(long *) realloc (a[y],(a[y][0]+1)*sizeof(long));
        a[y][a[y][0]]=x;
    }
    
    
    while (bfs()){
          min=INF;
          for (w=1;w<=nr;w++){
              min=c[sol[w]][n];
              
              for (i=sol[w];i!=1;i=tata[i])
                  if (c[tata[i]][i]<min) min=c[tata[i]][i];
              
              c[sol[w]][n]-=min;c[n][sol[w]]+=min;
              for (i=sol[w];i!=1;i=tata[i]) c[tata[i]][i]-=min,c[i][tata[i]]+=min;
          
              flux+=min;
          
          }    
    }
    
    printf("%d\n",flux);
    
    
    return 0;
}