poj 2112 最大流+floyd+二分答案

/*
题目描述：
k个机器，每个机器最多服务m头牛。
c头牛，每个牛需要1台机器来服务。
告诉你牛与机器每个之间的直接距离。
问：让所有的牛都被服务的情况下，使走的最远的牛的距离最短，求这个距离。
分析：
     首先我们可以用floyd求出各个点的最短距离d[i][j]！
然后构图：建图方式s->到每头牛建边容量为1，牛到机器建边容量为1，机器到t建边容量为m
如果，max_flow==c 那么 二分更新其 ans
二分的关键是 大于 ans 的边 不Add进去
*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
#define M 41000
const int inf=1<<29;
int gap[M],dis[M],pre[M],cur[M];
int NE,NV,sink,source,d[402][402];
int head[M],k,c,m,ans;
struct Node
{
    int c,pos,next;
} E[999999];
#define FF(i,NV) for(int i=0;i<NV;i++)
int sap(int s,int t)
{
    //memset(pre,-1,sizeof(pre));
    memset(dis,0,sizeof(int)*(NV+1));
    memset(gap,0,sizeof(int)*(NV+1));
    FF(i,NV) cur[i] = head[i];
    int u = pre[s] = s,maxflow = 0,aug = 1<<29;
    gap[0] = NV;
    while(dis[s] < NV)
    {
loop:
        for(int &i = cur[u]; i != -1; i = E[i].next)
        {
            int v = E[i].pos;
            if(E[i].c && dis[u] == dis[v] + 1)
            {
                aug=min(aug,E[i].c);
                pre[v] = u;
                u = v;
                if(v == t)
                {
                    maxflow += aug;
                    for(u = pre[u]; v != s; v = u,u = pre[u])
                    {
                        E[cur[u]].c -= aug;
                        E[cur[u]^1].c += aug;
                    }
                    aug = 1<<29;
                }

                goto loop;
            }
        }

        int mindis = NV;
        for(int i = head[u]; i != -1 ; i = E[i].next)
        {
            int v = E[i].pos;
            if(E[i].c && mindis > dis[v])
            {
                cur[u] = i;
                mindis = dis[v];
            }
        }
        if( (--gap[dis[u]]) == 0)break;
        gap[ dis[u] = mindis+1 ] ++;
        u = pre[u];
    }
    return maxflow;
}
void addEdge(int u,int v,int c )
{
    E[NE].c = c;
    E[NE].pos = v;
    E[NE].next = head[u];
    head[u] = NE++;
    E[NE].c = 0;
    E[NE].pos = u;
    E[NE].next = head[v];
    head[v] = NE++;
}
void floyd(int n)
{
    int i,j,k;
    for(k=1; k<=n; k++)
    {
        for(i=1; i<=n; i++)
        {
            if(d[i][k]!=inf)
                for(j=1; j<=n; j++)
                {
                    if(d[k][j]!=inf&&i!=j)
                    {
                        d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
                        ans=max(d[i][j],ans);
                    }
                }
        }
    }
}
int ok(int mid)
{
    int i,j;
    NE=0;
    memset(head,-1,sizeof(int)*(NV+1));
    for(i=1;i<=k;i++) addEdge(i,sink,m);
    for(i=k+1;i<=k+c;i++) addEdge(source,i,1);
    for(i=k+1;i<=k+c;i++)
    {
        for(j=1;j<=k;j++)
        {
            if(d[i][j]<=mid) addEdge(i,j,1);
        }
    }
    if(sap(source,sink)==c) return 1;
    return 0;
}
int main()
{
    int i,j;
    while(scanf("%d%d%d",&k,&c,&m)!=EOF)
    {
        source=ans=0;
        sink=k+c+1;
        NV=sink+1;
        for(i=1; i<=k+c; i++)
        {
            for(j=1; j<=k+c; j++)
            {
                scanf("%d",&d[i][j]);
                if(d[i][j]==0) d[i][j]=inf;
            }
        }
        floyd(k+c);
        int l=1,r=ans;
        while(l<=r)
        {
            int mid=(l+r)>>1;
            if(ok(mid)) r=mid-1;
            else l=mid+1;
        }
        printf("%d\n",l);
    }
    return 0;
}
/*
2 3 2
0 3 2 1 1
3 0 3 2 0
2 3 0 1 0
1 2 1 0 2
1 0 0 2 0

*/