Pagini recente » Cod sursa (job #1429205) | Cod sursa (job #577240) | Cod sursa (job #653393) | Cod sursa (job #99273) | Cod sursa (job #1161998)
// Flux maxim de cost minim - O(N^2 * M^2) - Bellman
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#define Nmax 355
#define oo 2000000000
#define pb push_back
using namespace std;
ifstream f("fmcm.in");
ofstream g("fmcm.out");
int N,M,Source,Sink,x,y,c,z,MaxFlow,CostMin,used[Nmax];
int cap[Nmax][Nmax],cost[Nmax][Nmax],flow[Nmax][Nmax],ante[Nmax],d[Nmax];
vector < int > G[Nmax];
bitset < Nmax > inQ;
queue < int > Q;
inline void ReadInput()
{
f>>N>>M>>Source>>Sink;
for(int i=1;i<=M;++i)
{
f>>x>>y>>c>>z;
G[x].pb(y) , G[y].pb(x);
cap[x][y]=c;
cost[x][y]=z , cost[y][x]=-z;
}
}
int Bellman(int Source)
{
for(int i=1;i<=N;++i)d[i]=oo , ante[i]=oo , used[i]=0;
d[Source]=0 , ante[Source]=Source , inQ[Source]=1;
for(Q.push(Source); !Q.empty();Q.pop())
{
int node=Q.front();
inQ[node]=0; ++used[node];
if(used[node]==N)return 0;
for(vector<int>::iterator it=G[node].begin();it!=G[node].end();++it)
if(d[*it]>d[node]+cost[node][*it] && flow[node][*it]<cap[node][*it])
{
d[*it]=d[node]+cost[node][*it];
ante[*it]=node;
if(!inQ[*it])Q.push(*it) , inQ[*it]=1;
}
}
return (d[Sink]!=oo);
}
void GetFMCM()
{
for( ; Bellman(Source) ; )
{
int MinFlow=oo;
for(int i=Sink; i!=Source ; i=ante[i])
MinFlow=min(MinFlow,cap[ante[i]][i]-flow[ante[i]][i]);
if(MinFlow)
{
for(int i=Sink; i!=Source ; i=ante[i])
{
flow[ante[i]][i]+=MinFlow;
flow[i][ante[i]]-=MinFlow;
}
MaxFlow+=MinFlow;
CostMin+=MinFlow*d[Sink];
}
}
}
int main()
{
ReadInput();
GetFMCM();
g<<CostMin<<'\n';
f.close();g.close();
return 0;
}