Cod sursa(job #1400372)

Utilizator andreimdvMoldovan Andrei andreimdv Data 25 martie 2015 11:32:20
Problema Flux maxim de cost minim Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.72 kb
#include <fstream>
#include<vector>
#include<queue>
using namespace std;

ifstream fin("fmcm.in");
ofstream fout("fmcm.out");

int n,m,start,finish,aux,a,b,c,d,fluxminimactual,sol,distmin[400],i;
int cost[400][400],capacitate[400][400],tata[400];
vector<int> v[400];
queue<int>q;

bool bellman()
{
    for(int i=1;i<=n;++i)
    {
        tata[i]=0;
        distmin[i]=1<<30;
    }
    distmin[start]=0;
    q.push(start);
    tata[start]=-1;
    while(!q.empty())
    {
        int x=q.front();
        q.pop();
        for(int i=0;i<(int)v[x].size();++i)
        {
            int y=v[x][i];
            if(distmin[y]>distmin[x]+cost[x][y]&&capacitate[x][y]>0)
            {
                tata[y]=x;
                distmin[y]=distmin[x]+cost[x][y];
                q.push(y);
            }
        }
    }
    return tata[finish];
}



int main()
{
    fin>>n>>m>>start>>finish;
    for(i=1;i<=m;++i)
    {
        fin>>a>>b>>c>>d;
        v[a].push_back(b);
        v[b].push_back(a);
        cost[a][b]=d;
        cost[b][a]=-d; // !!!
        capacitate[a][b]=c;
    }
    while(bellman()) //un drum de cost minim de la start la finish retinut in vectorul tata
    {
        fluxminimactual=capacitate[tata[finish]][finish];
        for(i=finish;tata[i]!=-1;i=tata[i])
        {
            fluxminimactual=min(fluxminimactual,capacitate[tata[i]][i]);
        }
        for(i=finish;tata[i]!=-1;i=tata[i])
        {
            sol+=cost[tata[i]][i]*fluxminimactual;
            capacitate[tata[i]][i]-=fluxminimactual;
            capacitate[i][tata[i]]+=fluxminimactual;
        }
        capacitate[finish][tata[finish]]-=fluxminimactual;
    }
    fout<<sol;
    return 0;
}