Cod sursa(job #1356116)

Utilizator goalexboxerFMI Alexandru Ionascu goalexboxer Data 23 februarie 2015 10:48:47
Problema Flux maxim de cost minim Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 2.55 kb
#include<iostream>
#include<fstream>
#include<vector>
#include<queue>
using namespace std;

#define MAXSIZE 355
#define FIN "fmcm.in"
#define FOUT "fmcm.out"

int n,m;
vector<int> graph[MAXSIZE];
int capacity[MAXSIZE][MAXSIZE];
int cost[MAXSIZE][MAXSIZE];
int maxFlow;
int minCost;
int father[MAXSIZE];
int source,destination;

int read()
{
    freopen(FIN,"r",stdin);
    freopen(FOUT,"w",stdout);

    scanf("%d %d %d %d \n", &n,&m, &source, &destination);
    int x,y,z,t;
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d %d %d \n", &x,&y,&z,&t);
        graph[x].push_back(y);
        capacity[x][y] = z;
        cost[x][y] = t;
    }

    return 0;
}

int BFS()
{
    queue<int> q;
    vector<int>::iterator it;

    father[1] = -1;
    for(int i=2;i<=n;i++)
        father[i] = 0;

    q.push(source);
    while(!q.empty())
    {
        int nod = q.front();
       // printf("%d ", q.front());
        if(nod == destination)
        {
            father[destination] = nod;
            int minim = cost[nod][destination];
            while(!q.empty())
            {
                q.pop();
                for(it = graph[q.front()].begin();it!=graph[q.front()].end();it++)
                {
                    if(*it == destination && capacity[*it][destination] > 0)
                    {
                        if(cost[*it][destination] < minim)
                        {
                            minim = cost[*it][destination];
                            father[destination] = *it;
                        }
                    }
                }
            }

            return 1;

        }
        else
        {
            q.pop();
            for(it = graph[nod].begin();it!=graph[nod].end();it++)
            {
                if(father[*it] == 0 && capacity[nod][*it] > 0)
                {
                    q.push(*it);
                    father[*it] = nod;
                }
            }
        }
    }
    return 0;
}

int solve()
{
    while(BFS() == 1)
    {
        //get min value
        int minn = capacity[father[destination]][destination];
        for(int i=destination;i!=source;i=father[i])
        {
            if(minn > capacity[father[i]][i])
                minn = capacity[father[i]][i];
        }

        for(int i=destination;i!=source;i=father[i])
        {
            minCost += capacity[father[i]][i] * cost[father[i]][i];
            capacity[father[i]][i] -= minn;
        }

        maxFlow += minn;
    }

    return 0;
}

int write()
{
    printf("%d",minCost);

    return 0;
}

int main()
{
    read();
    solve();
    write();

    return 0;
}