Cod sursa(job #604035)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 19 iulie 2011 21:25:06
Problema Flux maxim de cost minim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.96 kb
#include<stdio.h>
#include<string.h>
#define N 350
int n,m,k,c[N][N],f[N][N],p,z,i,s,d,pre[N],t,l,h[N],poz[N],j,g[N][N],deg[N],o[N][N],que[30000],q,a[N],b[N];
long e;
int main()
{freopen("fmcm.in","r",stdin),freopen("fmcm.out","w",stdout);
scanf("%d%d%d%d",&n,&m,&s,&d),s--,d--,memset(a,10001,sizeof(a)),a[s]=p=q=0,que[q++]=s;
while(m--)
          scanf("%d%d%d%d",&i,&j,&p,&z),i--,j--,g[i][deg[i]++]=j,g[j][deg[j]++]=i,o[i][j]=z,o[j][i]=-z,c[i][j]=p;
fclose(stdin);
while(p<q)
          {i=que[p++];
          for(j=0;j<deg[i];j++)
          if(f[i][g[i][j]]<c[i][g[i][j]]&&a[g[i][j]]>a[i]+o[i][g[i][j]])
                  a[g[i][j]]=a[i]+o[i][g[i][j]],que[q++]=g[i][j];}
for(i=0;i<n;i++)
for(j=0;j<deg[i];j++)
          o[i][g[i][j]]+=a[i]-a[g[i][j]];
while(1)      
          {memset(poz,-1,sizeof(poz)),memset(b,10001,sizeof(b)),b[s]=l=0,m=10001,h[l++]=s,pre[d]=-1;
          while(l)
                   {i=h[0],poz[i]=-1,h[0]=h[--l],poz[h[0]]=0,j=1;
                   while(j<l-1)
                   if(b[h[j+1]]<b[h[0]])
                          j++,p=h[j],h[j]=h[0],h[0]=p,poz[h[j]]=j,poz[h[0]]=0;
                   else
                          j<<=1;
                   for(k=0;k<deg[i];k++)
                   if(c[i][g[i][k]]>f[i][g[i][k]]&&b[g[i][k]]>b[i]+o[i][g[i][k]])
                               {b[g[i][k]]=b[i]+o[i][g[i][k]],pre[g[i][k]]=i,j=poz[g[i][k]];
                               if(poz[g[i][k]]==-1)
                                      j=poz[g[i][k]]=l,h[l++]=g[i][k];
                               for(t=j>>1;t&&b[t]>b[j];t>>=1)
                                      p=h[j],h[j]=h[t],h[t]=p,poz[h[j]]=j,poz[h[t]]=t;}}
          if(pre[d]==-1)
                   break;
          for(l=d;l!=s;l=pre[l])
          if(m>(t=c[pre[l]][l]-f[pre[l]][l]))
                   m=t;
          for(l=d;l!=s;l=pre[l])
                   f[pre[l]][l]+=m,f[l][pre[l]]-=m;
          e+=(b[d]+a[d])*m;}
printf("%ld",e),fclose(stdout);
return 0;}