Cod sursa(job #1645636)

Utilizator george_stelianChichirim George george_stelian Data 10 martie 2016 13:08:31
Problema Flux maxim de cost minim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 2.37 kb
#include <cstdio>
#include <vector>
#include <queue>

using namespace std;

const int inf=1000000000;
struct edge
{
    int nod,c;
    bool operator <(const edge &aux) const
    {
        return c>aux.c;
    }
};
vector<edge> v[355];
queue<int> q;
priority_queue<edge> h;
int dist[355],cap[355][355],flow[355][355],d[355],tata[355],sursa,dest,n;
char vaz[355];

void bellman_ford(int nod)
{
    for(int i=1;i<=n;i++) dist[i]=inf;
    dist[nod]=0;
    q.push(nod);
    vaz[nod]=1;
    while(!q.empty())
    {
        int nod=q.front();q.pop();
        vaz[nod]=0;
        for(vector<edge>::iterator it=v[nod].begin();it!=v[nod].end();it++)
            if(cap[nod][it->nod] && dist[it->nod]>dist[nod]+it->c)
            {
                dist[it->nod]=dist[nod]+it->c;
                if(!vaz[it->nod] && it->nod!=dest)
                {
                    vaz[it->nod]=1;
                    q.push(it->nod);
                }
            }
    }
}

int dijkstra()
{
    for(int i=1;i<=n;i++) d[i]=inf;
    d[sursa]=0;
    h.push({sursa,0});
    while(!h.empty())
    {
        int nod=h.top().nod,c=h.top().c;
        h.pop();
        if(d[nod]!=c) continue;
        for(vector<edge>::iterator it=v[nod].begin();it!=v[nod].end();it++)
            if(cap[nod][it->nod]>flow[nod][it->nod] && d[it->nod]>d[nod]+it->c)
            {
                d[it->nod]=d[nod]+it->c;
                tata[it->nod]=nod;
                if(it->nod!=dest) h.push({it->nod,d[it->nod]});
            }
    }
    return d[dest]!=inf;
}

int main()
{
    freopen("fmcm.in", "r", stdin);
    freopen("fmcm.out", "w", stdout);
    int m,x,y,a,c,sol=0;
    scanf("%d%d%d%d",&n,&m,&sursa,&dest);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d%d",&x,&y,&a,&c);
        v[x].push_back({y,c});
        v[y].push_back({x,-c});
        cap[x][y]=a;
    }
    bellman_ford(sursa);
    for(int i=1;i<=n;i++)
        for(int j=0;j<v[i].size();j++)
            v[i][j].c+=dist[i]-dist[v[i][j].nod];
    while(dijkstra())
    {
        int s=inf;
        for(int i=dest;i!=sursa;i=tata[i]) s=min(s,cap[tata[i]][i]-flow[tata[i]][i]);
        for(int i=dest;i!=sursa;i=tata[i])
        {
            flow[tata[i]][i]+=s;
            flow[i][tata[i]]-=s;
        }
        sol+=s*(d[dest]-dist[sursa]+dist[dest]);
    }
    printf("%d",sol);
    return 0;
}