費用流學習筆記

費用流學習筆記

最小/最大費用最大流

每一條邊不僅有流量還有費用的限制

要滿足流量最大的前提下費用儘可能小/大

增廣的時候選擇費用的最短路/最長路增廣就行了

建邊的時候反向邊的流量為 \(0\),費用變為相反數

因為費用有可能為負數,所以要拿 \(spfa\) 預處理出最短路/最長路

如果寫法是 \(EK\) 單路增廣的話需要一個 \(pre\) 數組來記錄最短路/最長路是從哪裡轉移來的

如果是 \(dinic\) 多路增廣則要注意在 \(dfs\) 的時候給每一個點打上一個 \(vis\) 標記,否則會在總費用為 \(0\) 的環中死循環

程式碼(EK)

#include<cstdio>
#include<iostream>
#include<queue>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstring>
#define rg register
inline int read(){
	rg int x=0,fh=1;
	rg char ch=getchar();
	while(ch<'0' || ch>'9'){
		if(ch=='-') fh=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9'){
		x=(x<<1)+(x<<3)+(ch^48);
		ch=getchar();
	}
	return x*fh;
}
const int maxn=1e5+5;
int h[maxn],tot=2,s,t,n,m;
struct asd{
	int to,nxt,val,cost;
}b[maxn<<1];
void ad(rg int aa,rg int bb,rg int cc,rg int dd){
	b[tot].to=bb;
	b[tot].nxt=h[aa];
	b[tot].val=cc;
	b[tot].cost=dd;
	h[aa]=tot++;
}
int dis[maxn],incf[maxn],pre[maxn];
bool inq[maxn];
std::queue<int> q;
bool spfa(){
	memset(dis,0x3f,sizeof(dis));
	memset(incf,0,sizeof(incf));
	incf[s]=0x3f3f3f3f;
	dis[s]=0;
	q.push(s);
	inq[s]=1;
	while(!q.empty()){
		rg int now=q.front();
		q.pop();
		inq[now]=0;
		for(rg int i=h[now];i!=-1;i=b[i].nxt){
			rg int u=b[i].to;
			if(b[i].val && dis[u]>dis[now]+b[i].cost){
				dis[u]=dis[now]+b[i].cost;
				pre[u]=i;
				incf[u]=std::min(incf[now],b[i].val);
				if(!inq[u]){
					inq[u]=1;
					q.push(u);
				}
			}
		}
	}
	return dis[t]!=0x3f3f3f3f;
}
int ans1,ans2;
void updat(){
	ans1+=incf[t];
	ans2+=dis[t]*incf[t];
	rg int now=t,i;
	while(now){
		i=pre[now];
		b[i].val-=incf[t];
		b[i^1].val+=incf[t];
		now=b[i^1].to;
	}
}
int main(){
	memset(h,-1,sizeof(h));
	n=read(),m=read(),s=read(),t=read();
	rg int aa,bb,cc,dd;
	for(rg int i=1;i<=m;i++){
		aa=read(),bb=read(),cc=read(),dd=read();
		ad(aa,bb,cc,dd);
		ad(bb,aa,0,-dd);
	}
	while(spfa()) updat();
	printf("%d %d\n",ans1,ans2);
	return 0;
}

程式碼(dinic)

#include<cstdio>
#include<iostream>
#include<queue>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstring>
#define rg register
inline int read(){
	rg int x=0,fh=1;
	rg char ch=getchar();
	while(ch<'0' || ch>'9'){
		if(ch=='-') fh=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9'){
		x=(x<<1)+(x<<3)+(ch^48);
		ch=getchar();
	}
	return x*fh;
}
const int maxn=1e5+5;
int h[maxn],tot=2,s,t,n,m;
struct asd{
	int to,nxt,val,cost;
}b[maxn<<1];
void ad(rg int aa,rg int bb,rg int cc,rg int dd){
	b[tot].to=bb;
	b[tot].nxt=h[aa];
	b[tot].val=cc;
	b[tot].cost=dd;
	h[aa]=tot++;
}
int dis[maxn];
bool inq[maxn];
std::queue<int> q;
bool spfa(){
	memset(dis,0x3f,sizeof(dis));
	dis[s]=0;
	q.push(s);
	inq[s]=1;
	while(!q.empty()){
		rg int now=q.front();
		q.pop();
		inq[now]=0;
		for(rg int i=h[now];i!=-1;i=b[i].nxt){
			rg int u=b[i].to;
			if(b[i].val && dis[u]>dis[now]+b[i].cost){
				dis[u]=dis[now]+b[i].cost;
				if(!inq[u]){
					inq[u]=1;
					q.push(u);
				}
			}
		}
	}
	return dis[t]!=0x3f3f3f3f;
}
int ans1,ans2;
int dfs(rg int now,rg int ac1){
	if(now==t){
		ans1+=ac1;
		return ac1;
	}
	inq[now]=1;
	rg int ac2=0;
	for(rg int i=h[now];i!=-1;i=b[i].nxt){
		rg int u=b[i].to;
		if(!inq[u] && b[i].val && dis[u]==dis[now]+b[i].cost){
			rg int nans=dfs(u,std::min(ac1,b[i].val));
			ac1-=nans,ac2+=nans,b[i].val-=nans,b[i^1].val+=nans;
			ans2+=nans*b[i].cost;
		}
		if(!ac1) break;
	}
	inq[now]=0;
	if(!ac2) dis[now]=0x3f3f3f3f;
	return ac2;
}
int main(){
	memset(h,-1,sizeof(h));
	n=read(),m=read(),s=read(),t=read();
	rg int aa,bb,cc,dd;
	for(rg int i=1;i<=m;i++){
		aa=read(),bb=read(),cc=read(),dd=read();
		ad(aa,bb,cc,dd);
		ad(bb,aa,0,-dd);
	}
	while(spfa()) dfs(s,0x3f3f3f3f);
	printf("%d %d\n",ans1,ans2);
	return 0;
}

最小/最大費用可行流

和最小/最大費用最大流的區別是不再需要滿足流量最大

前者有可能為了增廣流量而使得總費用變大/變小

所以要在每次求出最短/最長路之後判斷一下 \(dis[t]\) 是否仍然小於/大於 \(0\)

如果不滿足就不要繼續增廣了

Tags: