Cod sursa(job #1495376)

Utilizator alexpetrescuAlexandru Petrescu alexpetrescu Data 2 octombrie 2015 23:54:45
Problema Flux maxim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 2.13 kb
#include <cstdio>
#include <cstring>
#define INF 1000000000
#define MAXN 1000
#define MAXM 5000
int c[2*MAXM+1], f[2*MAXM+1], from[MAXN+1], n, q[MAXN+1], h, val[2*MAXM+1], next[2*MAXM+1], lista[MAXN+1], e[MAXN+1];
inline void adauga(int x, int y, int z){
    h++;
    c[h]=z;
    f[h]=0;
    val[h]=y;
    next[h]=lista[x];
    lista[x]=h;
}
inline bool bfs(){
    int st, dr, p;
    st=0;
    dr=1;
    q[0]=1;
    memset(from, 0, sizeof from);
    memset(e, 0, sizeof e);
    while(st<dr){
        for(p=lista[q[st]]; p; p=next[p]){
            if((val[p]!=1)&&(from[val[p]]==0)&&(c[p]>f[p])){
                from[val[p]]=q[st];
                e[val[p]]=p;
                q[dr++]=val[p];
            }
        }
        if(from[n]!=0){
            return 1;
        }
        st++;
    }
    return 0;
}
int main(){
    int m, i, x, y, z, ans, min, p, q;
    FILE *fin, *fout;
    fin=fopen("maxflow.in", "r");
    fout=fopen("maxflow.out", "w");
    fscanf(fin, "%d%d", &n, &m);
    for(i=1; i<=m; i++){
        fscanf(fin, "%d%d%d", &x, &y, &z);
        adauga(x, y, z);
        adauga(y, x, 0);
    }
    ans=0;
    while(bfs()){
        for(p=lista[n]; p; p=next[p]){
            if((from[val[p]])&&(c[p]-f[p]>0)){
                if(p%2==0){
                    q=p-1;
                }else{
                    q=p+1;
                }
                min=c[q]-f[q];
                i=val[p];
                while(from[i]){
                    if(min>c[e[i]]-f[e[i]]){
                        min=c[e[i]]-f[e[i]];
                    }
                    i=from[i];
                }
                f[q]+=min;
                f[p]-=min;
                i=val[p];
                while(from[i]){
                    f[e[i]]+=min;
                    if(e[i]%2==0){
                        f[e[i]-1]-=min;
                    }else{
                        f[e[i]+1]-=min;
                    }
                    i=from[i];
                }
                ans+=min;
            }
        }
    }
    fprintf(fout, "%d\n", ans);
    fclose(fin);
    fclose(fout);
    return 0;
}