2018.10.14 loj#6012. 「网络流 24 题」分配问题(费用流)

传送门

费用流水题。


依然是照着题意模拟建边就行了。

为了练板子又重新写了一遍费用流。

代码:

#include<bits/stdc++.h>
#define N 305
#define M 90005
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
int n;
struct edge{int v,next,c,w;};
struct MCMF{
	int first[N],d[N],pred[N],pos[N],flow[N],s,t,cnt;
	edge e[M];
	bool in[N];
	inline void init(){memset(first,-1,sizeof(first)),cnt=-1,s=0,t=n*2+1;}
	inline void addedge(int u,int v,int c,int w){e[++cnt].v=v,e[cnt].c=c,e[cnt].w=w,e[cnt].next=first[u],first[u]=cnt;}
	inline void add(int u,int v,int c,int w){addedge(u,v,c,w),addedge(v,u,0,-w);}
	inline bool spfa1(){
		queue<int>q;
		for(int i=1;i<=t;++i)d[i]=0x3f3f3f3f;
		d[s]=0,pred[t]=-1,flow[s]=0x3f3f3f3f,in[s]=1,q.push(s);
		while(!q.empty()){
			int x=q.front();
			q.pop(),in[x]=false;
			for(int i=first[x];~i;i=e[i].next){
				int v=e[i].v;
				if(e[i].c&&d[v]>d[x]+e[i].w){
					d[v]=d[x]+e[i].w,flow[v]=min(flow[x],e[i].c),pred[v]=x,pos[v]=i;
					if(!in[v])in[v]=1,q.push(v);
				}
			}
		}
		return d[t]!=0x3f3f3f3f;
	}
	inline int solve1(){
		int ret=0;
		for(int w=t;spfa1();w=t){
			ret+=d[t];
			while(w!=s)e[pos[w]].c-=flow[t],e[pos[w]^1].c+=flow[t],w=pred[w];
		}
		return ret;
	}
	inline bool spfa2(){
		queue<int>q;
		for(int i=1;i<=t;++i)d[i]=-0x3f3f3f3f;
		d[s]=0,pred[t]=-1,flow[s]=0x3f3f3f3f,in[s]=1,q.push(s);
		while(!q.empty()){
			int x=q.front();
			q.pop(),in[x]=false;
			for(int i=first[x];~i;i=e[i].next){
				int v=e[i].v;
				if(e[i].c&&d[v]<d[x]+e[i].w){
					d[v]=d[x]+e[i].w,flow[v]=min(flow[x],e[i].c),pred[v]=x,pos[v]=i;
					if(!in[v])in[v]=1,q.push(v);
				}
			}
		}
		return d[t]!=-0x3f3f3f3f;
	}
	inline int solve2(){
		int ret=0;
		for(int w=t;spfa2();w=t){
			ret+=d[t];
			while(w!=s)e[pos[w]].c-=flow[t],e[pos[w]^1].c+=flow[t],w=pred[w];
		}
		return ret;
	}
}mcmf1,mcmf2;
int main(){
	n=read(),mcmf1.init(),mcmf2.init();
	for(int i=1;i<=n;++i)mcmf1.add(mcmf1.s,i,1,0),mcmf1.add(i+n,mcmf1.t,1,0),mcmf2.add(mcmf2.s,i,1,0),mcmf2.add(i+n,mcmf2.t,1,0);
	for(int val,i=1;i<=n;++i)for(int j=1;j<=n;++j)val=read(),mcmf1.add(i,j+n,1,val),mcmf2.add(i,j+n,1,val);
	cout<<mcmf1.solve1()<<'\n'<<mcmf2.solve2();
	return 0;
}
上一篇:正则表达式写法:Java和Js比较


下一篇:Vue学习笔记-Vue.js-2.X 学习(四)===>脚手架Vue-CLI(基本工作和创建)