POJ2112: Optimal Milking 题解

首先跑一个floyd得到任意奶牛到任意挤奶器的最短距离

刚开始以为求所有奶牛路程和的最小值，于是跑了一个费用流，WA

然后发现求的是最远的奶牛的路程的最小值。。。

于是可以二分答案啊

但是对于网络流题目来说不必二分答案，可以直接枚举答案，对于答案i，将距离正好是i的点对连边，这样可以直接用ans=i-1时候的残余网络再跑最大流，答案直接叠加就行

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cmath>
#include <algorithm>
#include <cstdlib>
#include <utility>
#include <map>
#include <stack>
#include <set>
#include <vector>
#include <queue>
#include <deque>
#define x first
#define y second
#define mp make_pair
#define pb push_back
#define LL long long
#define Pair pair<int,int>
#define LOWBIT(x) x & (-x)
using namespace std;

const int MOD=1e9+7;
const int INF=0x7ffffff;
const int magic=348;

vector<Pair> v[500048];

int t,tot=1,head[100048],nxt[200048],to[200048],f[200048];
inline void addedge(int s,int t,int cap)
{
	to[++tot]=t;nxt[tot]=head[s];head[s]=tot;f[tot]=cap;
	to[++tot]=s;nxt[tot]=head[t];head[t]=tot;f[tot]=0;
}

int n,m,lim;
int ga[301][301];
void floyd()
{
	int i,j,k;
	for (k=1;k<=n+m;k++)
		for (i=1;i<=n+m;i++)
			for (j=1;j<=n+m;j++)
				if (i!=j && i!=k && j!=k && ga[i][j]>ga[i][k]+ga[k][j])
					ga[i][j]=ga[i][k]+ga[k][j];
}

int depth[100048];queue<int> q;

bool bfs()
{
	int i,x,y;
	for (i=0;i<=t;i++) depth[i]=-1;
	depth[0]=0;
	q.push(0);
	while (!q.empty())
	{
		x=q.front();q.pop();
		for (i=head[x];i;i=nxt[i])
		{
			y=to[i];
			if (f[i] && depth[y]==-1)
			{
				depth[y]=depth[x]+1;
				q.push(y);
			}
		}
	}
	if (depth[t]==-1) return false; else return true;
}

int dfs(int x,int maxf)
{
	if (x==t) return maxf;
	int i,y,ans=0,now,minf;
	for (i=head[x];i;i=nxt[i])
	{
		y=to[i];
		if (f[i] && depth[y]==depth[x]+1)
		{
			minf=min(f[i],maxf-ans);
			now=dfs(y,minf);
			f[i]-=now;
			f[i^1]+=now;
			ans+=now;
		}
	}
	return ans;
}

int main ()
{
	int i,j;
	scanf("%d%d%d",&n,&m,&lim);
	t=n+m+1;
	for (i=1;i<=n+m;i++)
		for (j=1;j<=n+m;j++)
		{
			scanf("%d",&ga[i][j]);
			if (i!=j && !ga[i][j]) ga[i][j]=INF;
		}
	floyd();
	/*for (i=1;i<=n;i++)
	{
		for (j=1;j<=m;j++)
			printf("%d ",ga[i][n+j]);
		cout<<endl;
	}*/
	for (i=1;i<=n;i++)
		for (j=1;j<=m;j++)
			if (ga[i][n+j]<INF) v[ga[i][n+j]].pb(mp(i,n+j));
	for (i=1;i<=n;i++) addedge(0,i,lim);
	for (i=1;i<=m;i++) addedge(n+i,t,1);
	int ans=0,max_flow=0;
	do
	{
		ans++;
		for (i=0;i<v[ans].size();i++) addedge(v[ans][i].x,v[ans][i].y,1);
		while (bfs()) max_flow+=dfs(0,2e9);
	}
	while (max_flow<m);
	printf("%d\n",ans);
	return 0;
}