Cod sursa(job #699777)

Utilizator mihai995mihai995 mihai995 Data 29 februarie 2012 21:18:17
Problema Flux maxim de cost minim Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include <fstream>
#include <vector>
#include <queue>
using namespace std;

const long long N=351,inf=(long long)1<<61;
long long cap[N][N],cost[N][N],flux[N][N],dist[N],T[N],n;
queue<int> Q;
vector<int> a[N];

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

inline long long dif(long long x,long long y)
{
    return cap[x][y]-flux[x][y];
}

bool bf(long long x,long long D)
{
    for (long long i=1;i<=n;i++)
        dist[i]=inf;
    while (!Q.empty())
        Q.pop();
    dist[x]=0;
    Q.push(x);
    while (!Q.empty())
    {
        x=Q.front();
        if (x==D)
            return true;
        Q.pop();
        for (vector<int>::iterator i=a[x].begin();i!=a[x].end();i++)
            if (dif(x,*i)>0 && dist[x]+cost[x][*i]<dist[*i])
            {
                dist[*i]=dist[x]+cost[x][*i];
                T[*i]=x;
                Q.push(*i);
            }
    }
    return false;
}

long long flow(long long S,long long D)
{
    long long C=0,M;
    while (bf(S,D))
    {
        M=inf;
        for (long long i=D;i!=S;i=T[i])
            M=min(M,dif(T[i],i));
        for (long long i=D;i!=S;i=T[i])
        {
            C+=cost[T[i]][i]*M;
            flux[T[i]][i]+=M;
            flux[i][T[i]]-=M;
        }
    }
    return C;
}

int main()
{
    long long m,x,y,c,z,S,D;
    in>>n>>m>>S>>D;
    while (m--)
    {
        in>>x>>y>>c>>z;
        a[x].push_back(y);
        a[y].push_back(x);
        cap[x][y]=c;
        cost[x][y]=z;
        cost[y][x]=-z;
    }
    out<<flow(S,D)<<"\n";
    return 0;
}