Cod sursa(job #526812)

Utilizator DraStiKDragos Oprica DraStiK Data 29 ianuarie 2011 15:26:10
Problema Flux maxim de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.02 kb
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;

#define pb push_back
#define mp make_pair
#define INF 0x3f3f
#define sc second
#define fs first
#define DIM 355

vector <pair <short,short> > g[DIM];
short c[DIM][DIM],f[DIM][DIM];
short dst[DIM],t[DIM];
short n,m,SRC,DST;
bool viz[DIM];
int cost;

struct cmp
{
    bool operator () (const short &a,const short &b)
    {
        return dst[a]>dst[b];
    }
}; priority_queue <short,vector <short>,cmp> q;

void read ()
{
    short i,x,y,z,t;

    scanf ("%hd%hd%hd%hd",&n,&m,&SRC,&DST);
    for (i=1; i<=m; ++i)
    {
        scanf ("%hd%hd%hd%hd",&x,&y,&z,&t);
        g[x].pb (mp (y,t));
        g[y].pb (mp (x,-t));
        c[x][y]=z;
    }
}

short dijkstra ()
{
    vector <pair <short,short> > :: iterator it;
    short nod;

    memset (t,0,sizeof (t));
    memset (viz,0,sizeof (viz));
    memset (dst,INF,sizeof (dst));
    dst[SRC]=0;
    for (q.push (SRC); !q.empty (); )
    {
        viz[nod=q.top ()]=0; q.pop ();
        for (it=g[nod].begin (); it!=g[nod].end (); ++it)
            if (dst[nod]+it->sc<dst[it->fs] && c[nod][it->fs]!=f[nod][it->fs])
            {
                dst[it->fs]=dst[nod]+it->sc;
                t[it->fs]=nod;
                if (!viz[it->fs])
                {
                    viz[it->fs]=1;
                    q.push (it->fs);
                }
            }
    }

    return dst[DST]!=INF;
}

void solve ()
{
    vector <short> :: iterator it;
    short i,min_cur;

    for ( ; dijkstra (); )
    {
        min_cur=INF;
        for (i=DST; i!=SRC; i=t[i])
            min_cur=min (min_cur,(short)(c[t[i]][i]-f[t[i]][i]));
        for (i=DST; i!=SRC; i=t[i])
        {
            f[t[i]][i]+=min_cur;
            f[i][t[i]]-=min_cur;
        }
        cost+=min_cur*dst[DST];
    }
    printf ("%d",cost);
}

int main ()
{
    freopen ("fmcm.in","r",stdin);
    freopen ("fmcm.out","w",stdout);

    read ();
    solve ();

    return 0;
}