Cod sursa(job #596166)

Utilizator BlaugranasEnal Gemaledin Blaugranas Data 16 iunie 2011 10:17:05
Problema Cuplaj maxim de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.38 kb
#include<stdio.h>
#define E 50001
#define N 606
int n,m,c[N][N]={0},f[N][N]={0},z,i,j,a[E],b[E],pre[N],p,q,t,l,nr=0;
long k,v[N][N],e=0,x[E],que[E],g[N];
int main()
{freopen("cmcm.in","r",stdin);
freopen("cmcm.out","w",stdout);
scanf("%d%d%ld",&n,&m,&k);     
for(i=1;i<=k;i++)
     {scanf("%d%d%d",&a[i],&b[i],&z);
     c[a[i]][b[i]+n+1]=1,v[a[i]][b[i]+n+1]=z,v[b[i]+n+1][a[i]]=-z;}
for(i=1;i<=n;i++)
     v[0][i]=0,c[0][i]=1;
for(i=n+2;i<=n+m+1;i++)
     v[i][n+1]=0,c[i][n+1]=1;
while(1)      
       {for(i=1;i<=n+m+1;i++)
               pre[i]=0,g[i]=E;
       g[0]=p=q=0;
       que[q++]=0;
       while(p<q)
               {t=que[p++];
               if(t&&t<=n)
                       {for(i=n+1;i<=n+m+1;i++)
                       if(f[t][i]<c[t][i]&&g[i]>g[t]+v[t][i])
                               que[q++]=i,pre[i]=t,g[i]=g[t]+v[t][i];}
               else
                       {for(i=1;i<=n+1;i++)
                       if(f[t][i]<c[t][i]&&g[i]>g[t]+v[t][i])
                               que[q++]=i,pre[i]=t,g[i]=g[t]+v[t][i];}}
       if(!pre[n+1])
               break;
       for(l=n+1;l;l=pre[l])
               f[pre[l]][l]++,f[l][pre[l]]--;
       e+=g[n+1];}
for(i=1;i<=k;i++)
if(f[a[i]][b[i]+n+1]==1)
       x[++nr]=i;
printf("%d %ld\n",nr,e);
for(i=1;i<=nr;i++)
       printf("%ld ",x[i]);    
fclose(stdin);
fclose(stdout);
return 0;}