Cod sursa(job #2633407)

Utilizator Leonard123Mirt Leonard Leonard123 Data 7 iulie 2020 14:22:15
Problema Flux maxim de cost minim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
	
#include <bits/stdc++.h>
using namespace std;
#define st first
#define nd second
#define pb push_back
#define mkp make_pair
#define FOR(i,a,b) for(int i=(a);i<=(b);++i)
#define FORS(i,a,b) for(int i=(a);i<(b);++i)
#define PII pair<int,int>
#define VI vector<int>
#define VPII vector<PII>
#define maxn 355
const int INF=0x3f3f3f3f;
 
int N,M,S,D,cst[maxn][maxn],C[maxn][maxn],nod,mini,
TT[maxn],Cmin,d[maxn],cost,reald[maxn],old[maxn];
VI g[maxn];
priority_queue<PII,VPII,greater<PII>>heap;
 
bool fmcm()
{
	FOR(i,1,N)d[i]=INF;
	d[S]=0;reald[S]=0;
	heap.push(mkp(d[S],S));
	while(!heap.empty())
	{
		cost=heap.top().st;
		nod=heap.top().nd;
		heap.pop();
		if(cost>d[nod]) continue;
		for(auto it:g[nod])
			if(C[nod][it]){
				int newc=d[nod]+cst[nod][it]+old[nod]-old[it];
				if(newc<d[it]){
					d[it]=newc;
					reald[it]=reald[nod]+cst[nod][it];
					TT[it]=nod;
					heap.push(mkp(newc,it));
				}
			}
	}
	memcpy(old,reald,sizeof(d));
	if(d[D]==INF)return 0;
	mini=INF;
	for(int u=D;u!=S;u=TT[u])
		mini=min(mini,C[TT[u]][u]);
	for(int u=D;u!=S;u=TT[u])
		C[TT[u]][u]-=mini,
		C[u][TT[u]]+=mini;
	Cmin+=(mini*reald[D]);
	return 1;
}
 
int main()
{
	freopen("fmcm.in","r",stdin);
	freopen("fmcm.out","w",stdout);
	int x,y,c,z;
	cin>>N>>M>>S>>D;
	FOR(i,1,M){
		cin>>x>>y>>c>>z;
		g[x].pb(y);
		g[y].pb(x);
		C[x][y]=c;
		cst[x][y]=z,cst[y][x]=-z;
	}
	for(;fmcm(););
	cout<<Cmin;
}