Cod sursa(job #766074)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 10 iulie 2012 11:24:12
Problema Flux maxim de cost minim Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.51 kb
#include<cstdio>
int n,m,k,j,i,l,s,d,t,q,w,c[351][351],p[351],h[351],r[351],g[351][351],a[351],o[351][351],b[351],e;
int main()
{freopen("fmcm.in","r",stdin);
freopen("fmcm.out","w",stdout);
scanf("%d%d%d%d",&n,&m,&s,&d);
while(m--)
          scanf("%d%d%d%d",&i,&j,&l,&k),g[i][a[i]++]=j,g[j][a[j]++]=i,o[i][j]=k,o[j][i]=-k,c[i][j]=l;
while(1)      
          {for(i=1;i<=n;i++)
                   b[i]=10001,h[i]=p[i]=i,r[i]=0;
          b[s]=p[d]=0,h[1]=r[1]=s,l=n,h[s]=r[s]=1;
          while(l)
                   {for(k=0,j=g[h[1]][0];k<a[h[1]];j=g[h[1]][++k])
                   if(b[j]>b[h[1]]+o[h[1]][j]&&c[h[1]][j])
                          {b[j]=b[h[1]]+o[h[1]][j],p[j]=h[1];
                          for(t=r[j];t>3&&b[h[t]]<b[h[q=t/2]];t=q)
                                 w=h[t],h[t]=h[q],h[q]=w,r[h[t]]=t,r[h[q]]=q;}
                   h[1]=h[l--],r[h[1]]=1;
                   if(l>1)
                          for(j=1;(t=2*j)<=l;j=t)
                                 {if(t<l&&b[h[t+1]]<b[h[t]])
                                       t++;
                                 if(b[h[j]]<=b[h[t]])
                                       break;
                                 k=h[j],h[j]=h[t],h[t]=k,r[h[j]]=j,r[h[t]]=t;}}
          if(!p[d])
                   break;
          for(m=10001,l=d;l!=s;l=p[l])
          if(m>c[p[l]][l])
                   m=c[p[l]][l];
          for(l=d;l!=s;l=p[l])
                   c[p[l]][l]-=m,c[l][p[l]]+=m;
          e=e+b[d]*m;}
printf("%d",e);
return 0;}