Cod sursa(job #1233176)

Utilizator alex_bucevschiBucevschi Alexandru alex_bucevschi Data 24 septembrie 2014 21:28:20
Problema Flux maxim de cost minim Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 2.21 kb
#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
#define N 355
#define oo 1<<30
using namespace std;
vector<int>v[N];
deque<int> Q;
priority_queue<pair<int,int> >q;
int n,m,s,d,CD,x,y,C[N][N],D[N][N],FL[N][N],db[N],tata[N],dr[N],dd[N],viz[N],i,X,sol;
void bellman()
{
    Q.push_back(s);
    for(i=1;i<=n;i++)
        db[i]=oo;
    db[s]=0;
    viz[s]=1;
    while(Q.size())
    {
        X=Q.front();
        Q.pop_front();
        viz[X]=0;
        for(vector<int>::iterator it=v[X].begin();it!=v[X].end();it++)
            if(C[X][*it]-FL[X][*it]>0&&db[*it]>db[x]+D[x][*it])
            {
                db[*it]=db[x]+D[x][*it];
                if(!viz[*it]){
                        Q.push_back(*it);
                        viz[*it]=1;
                }
            }

    }
}
bool dijkstra()
{
    q.push(make_pair(0,s));
    for(i=1;i<=n;i++)
    {
        tata[i]=0;
        dd[i]=oo;
    }
    dd[s]=0;
    while(q.size())
    {
        X=q.top().second;
        q.pop();
        for(vector<int>::iterator it=v[X].begin();it!=v[X].end();it++)
        {
            if(C[X][*it]-FL[X][*it]>0)
                if(dd[*it]>dd[X]+D[X][*it]+db[X]-db[*it])
                {
                    dd[*it]=dd[X]+D[X][*it]+db[X]-db[*it];
                    dr[*it]=dr[X]+D[X][*it];
                    q.push(make_pair(-dd[*it],*it));
                    tata[*it]=X;
                }
        }
    }
    for(i=1;i<=n;i++) db[i]=dr[i];
    return tata[d]?1:0;
}
void upd()
{

    int x=tata[d],y=d,fl;
    fl=C[x][y]-FL[x][y];;
    for(;y!=s;y=tata[y],x=tata[x])
        fl=min(fl,C[x][y]-FL[x][y]);
    for(x=tata[d],y=d;y!=s;y=tata[y],x=tata[x])
    {
        FL[x][y]+=fl;
        FL[y][x]-=fl;
        //sol+=fl*D[x][y];
    }
    sol+=db[d]*fl;

}
int main()
{
    freopen("fmcm.in","r",stdin);
    freopen("fmcm.out","w",stdout);
    scanf("%d%d%d%d",&n,&m,&s,&d);
    for(;m;m--)
    {
        scanf("%d%d",&x,&y);
        scanf("%d%d",&C[x][y],&D[x][y]);
        v[x].push_back(y);
        v[y].push_back(x);
        D[y][x]=-D[x][y];
    }
    bellman();
    while(dijkstra())
        upd();
    printf("%d\n",sol);
    return 0;
}