Cod sursa(job #374432)

Utilizator bent_larsenSturzu Antonio-Gabriel bent_larsen Data 17 decembrie 2009 00:29:19
Problema Flux maxim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.66 kb
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#define pb push_back
#define mp make_pair
 
using namespace std;
 
vector <int> v[1001];
int cap[1001][1001],prev[1001],n;
 
int bf()
{
    int coada[1000],viz[1001],tail,head,nod,i,nod1;
 
    tail=0;
    head=1;
    coada[tail]=1;//bag sursa in coada
    memset(viz,0,sizeof(viz));
    viz[1]=1;//marchez sursa ca vizitata
 
    while(tail!=head)
    {
        nod=coada[tail];
        tail++;
        for(i=0;i<(int) v[nod].size();i++)
        {
            nod1=v[nod][i];
            if(viz[nod1]==0 && cap[nod][nod1]>0)
            {
                viz[nod1]=1;
                coada[head++]=nod1;
                prev[nod1]=nod;
		if(nod1==n)
			return 1;
            }
        }
    }
   return 0;
}
                             
int flux_maxim()
{
    int flux=0,cmin=110001,nod=n;
 
       while(bf())
       {       
	        cmin=110001;
	        nod=n;
		while(nod!=1)
		{
			if(cap[prev[nod]][nod]<cmin)
				cmin=cap[prev[nod]][nod];
			nod=prev[nod];
		}
		nod=n;
		while(nod!=1)
		{
			cap[prev[nod]][nod]-=cmin;
			cap[nod][prev[nod]]+=cmin;
			nod=prev[nod];
		}
		flux+=cmin;
	}		
     return flux;
}
                                     
int main()
{
    int m,x,y,c,i;
    FILE *f=fopen("maxflow.in","r");
 
    fscanf(f,"%i%i",&n,&m);
    memset(cap,0,sizeof(cap));	
    for(i=0;i<m;i++)
    {
        fscanf(f,"%i%i%i",&x,&y,&c);
        v[x].pb(y);
        v[y].pb(x);
        cap[x][y]=c;
    }
    fclose(f);
    f=fopen("maxflow.out","w");
    fprintf(f,"%i\n",flux_maxim());
    fclose(f);
    return 0;
}