Cod sursa(job #597988)

Utilizator crushackPopescu Silviu crushack Data 24 iunie 2011 12:13:39
Problema Flux maxim de cost minim Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.7 kb
#include <fstream>
#include <stdio.h>
#include <string.h>
#include <vector>
#include <queue>
#include <list>
#include <bitset>
#include <algorithm>
#define NMax 400
#define INF 0x3f3f
using namespace std;

const char IN[]="fmcm.in",OUT[]="fmcm.out";

int N,M,S,D;
short C[NMax][NMax] , F[NMax][NMax] , Cost[NMax][NMax] , T[NMax] , P[NMax];
//bool visit[NMax];
vector<short> ad[NMax];

struct cmp{
	bool operator()(const int &a,const int &b) const{
		return T[a]>T[b];
	}
};

priority_queue< int , vector<int> , cmp > qu;

int Dijkstra()
{
	int x,i;
	bitset <NMax> visit;
	qu.push(S);
	//memset(visit,0,sizeof(visit));
	memset(T,0x3f,sizeof(T));
	
	T[S]=0;P[S]=S;
	while (!qu.empty())
	{
		x=qu.top();qu.pop();//visit[x]=false;
		
		for (i=0; i<(int)ad[x].size() ; ++i)
			if (C[x][ad[x][i]]>F[x][ad[x][i]] && T[x]+Cost[x][ad[x][i]]<T[ad[x][i]])
			{
				T[ad[x][i]]=T[x]+Cost[x][ad[x][i]];P[ad[x][i]]=x;
				if (!visit[ad[x][i]])
					qu.push(ad[x][i]),visit[ad[x][i]]=true;
			}
	}
	
	return T[D]!=INF;
}

int flux()
{
	int ret=0,i;
	
	while (Dijkstra())
	{
		int Fmin=INF;
		
		for (i=D;i!=S;i=P[i])
			Fmin=min(Fmin,C[P[i]][i]-F[P[i]][i]);
		
		if (Fmin==0) continue;
		
		for (i=D;i!=S;i=P[i])
			F[P[i]][i]+=Fmin,
			F[i][P[i]]-=Fmin;
			
		ret+= Fmin*T[D];
	}
	return ret;
}

int main()
{
	int x,y,c,z;
	//ifstream fin(IN);
	freopen(IN,"r",stdin);
	scanf("%d%d%d%d",&N,&M,&S,&D);
	while (M--)
	{
		scanf("%d%d%d%d",&x,&y,&c,&z);
		ad[x].push_back(y);
		ad[y].push_back(x);
		Cost[x][y]=z;
		Cost[y][x]=-z;
		C[x][y]=c;//C[y][x]=c;
	}
	fclose(stdin);
	
	freopen(OUT,"w",stdout);
	printf("%d\n",flux());
	fclose(stdout);
	
	return 0;
}