POJ 2112 Optimal Milking 二分+最大流

点击打开链接

题意:c头牛 k台机器 每台机器可以装m只牛,知道任意的牛-机器,机器-机器,牛-牛的距离
c<=200,k<=30,m<=15,问把n头牛装到k台机器中,使走的距离最远的牛,其距离尽量小?
建图:源点到牛:容量为1,费用为0.牛到机器容量为1,费用为最短距离.机器到汇点容量为m,费用为0.
则把c头牛装到k台机器的最小花费为:流量为c的最小费用流 
现在要求的是:最长路径最短,二分答案x,建图时忽略花费>x的路径,判断最大流是否能到达c即可. 

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#include <cstdio>
#include <map>
#include <vector>
using namespace std;
const int N=2e3+20;
const int inf=2e8;
int k,c,m,n;
int num,head[N];
int a[N][N],d[N][N];
struct node{
	int to,next,vol,cost;
}e[N*N];
void floyd()
{
	for(int p=1;p<=n;p++)//中间结点编号1~p 
	{
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				a[i][j]=min(a[i][j],a[i][p]+a[p][j]);
			}	
		}
	}
}
void insert(int u,int v,int vol,int cost)
{
	e[num].to=v,e[num].vol=vol,e[num].cost=cost;
	e[num].next=head[u],head[u]=num++;
	
	e[num].to=u,e[num].vol=0,e[num].cost=-cost;
	e[num].next=head[v],head[v]=num++;
}
void build(int x)
{
	memset(head,-1,sizeof(head));
	num=0;
	for(int i=k+1;i<=n;i++)
	{
		insert(0,i,1,0);//源点-cow 
		for(int j=1;j<=k;j++)
		{
			if(a[i][j]<=x)
				insert(i,j,1,a[i][j]);	
		}	
	}	
	for(int i=1;i<=k;i++)
		insert(i,n+1,m,0);
}
queue<int> q;
int inq[N],dist[N],pre[N],path[N];
bool SPFA(int s,int t,int x)
{
	memset(dist,0x3f,sizeof(dist));
	memset(pre,-1,sizeof(pre));
	memset(inq,0,sizeof(inq));	
	q.push(s),dist[s]=0,inq[s]=1;
	while(!q.empty())
	{
		int u=q.front();
		q.pop(),inq[u]=0;
		for(int i=head[u];i!=-1;i=e[i].next)
		{
			int v=e[i].to;
			if(e[i].vol>0 && dist[v]>dist[u]+e[i].cost)
			{
				dist[v]=dist[u]+e[i].cost;
				pre[v]=u,path[v]=i;
				if(!inq[v])
					q.push(v),inq[v]=1;		
			}
		}
	}
	if(pre[t]==-1)
		return false;
	return true;	
}
 
int Max_Flow(int s,int t,int x)
{
	int cost=0,flow=0;
	while(SPFA(s,t,x))
	{
		int f=inf;
		for(int u=t;u!=s;u=pre[u])
			f=min(f,e[path[u]].vol);
		flow+=f,cost+=dist[t]*f;
		for(int u=t;u!=s;u=pre[u])
		{
			e[path[u]].vol-=f;
			e[path[u]^1].vol+=f;
		}
	}
	return flow;
}
bool check(int x)
{
	int f=Max_Flow(0,n+1,x); 
	return f==c;
}
int main()
{
	while(cin>>k>>c>>m)
	{
		n=k+c;
		num=0;
		memset(head,-1,sizeof(head));	
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				scanf("%d",&a[i][j]);
				if(a[i][j]==0)
					a[i][j]=inf;//到不了为inf 
			}
		}
		floyd();//
		int l=0,r=5e4,ans;
		while(l<=r)
		{
			int mid=(l+r)>>1;
			build(mid);//忽略费用大于mid的边 
			if(check(mid))
			{
				ans=mid;
				r=mid-1;
			}
			else
				l=mid+1;
		}
		cout<<ans<<endl;
	}
	return 0;
}