[Poj2112][USACO2003 US OPEN] Optimal Milking [网络流，最大流][Dinic+当前弧优化]

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>
#include <queue>

using namespace std;

template<const int _n>
struct Edge
{
    struct Edge_base { int    to,next,w; }e[_n];
    int    p[_n],cnt;
    void    insert(const int x,const int y,const int z)
    { e[++cnt].to=y; e[cnt].next=p[x]; e[cnt].w=z; p[x]=cnt; return ; }
    int    start(const int x) { return p[x]; }
    void    clear() { cnt=1,memset(p,0,sizeof(p)); return ; }
    Edge_base&    operator[](const int x) { return e[x]; }
};

int    n,cur[21000],SSS,TTT;
int    level[21000],K,C,M,Map[310][310];
Edge<21000>    e;

bool    Bfs(const int S)
{
    int    i,t;
    queue<int>    Q;
    memset(level,0,sizeof(level));

    level[S]=1;
    Q.push(S);
    while(!Q.empty())
    {
        t=Q.front();Q.pop();
        for(i=e.start(t);i;i=e[i].next)
        {
            if(!level[e[i].to] && e[i].w)
            {
                level[e[i].to]=level[t]+1;
                Q.push(e[i].to);
            }
        }
    }
    return level[TTT];
}

int    Dfs(const int S,const int bk)
{
    if(S==TTT)return bk;
    int    rest=bk;
    for(int &i=cur[S];i;i=e[i].next)//这里地址（引用）符一定要加，这样cur才会变化，优化才有效
    {
        if(e[i].w && level[e[i].to]==level[S]+1)
        {
            int    flow=Dfs(e[i].to,min(rest,e[i].w));
            e[i].w-=flow;
            e[i^1].w+=flow;
            if((rest-=flow)==0)break;
        }
    }
    if(rest==bk)level[S]=0;
    return bk-rest;
}

int    Dinic()
{
    int    flow=0,i;
    while(Bfs(SSS))
    {
        for(i=1;i<=n;++i)cur[i]=e.start(i);//当前弧优化
　　　　　//据说这道题不加这个优化会变快。。可能是因为多了个for。。。
        flow+=Dfs(SSS,0x3f3f3ff);
    }
    return flow;
}

bool    Check(const int lim)
{
    int    i,j;
    e.clear();
    for(i=1;i<=K;++i)
    {
        e.insert(SSS,i,M);
        e.insert(i,SSS,0);
        for(j=K+1;j<=K+C;++j)
        {
            if(Map[i][j]<=lim)
            {
                e.insert(i,j,1);
                e.insert(j,i,0);
            }
        }
    }
    for(i=K+1;i<=C+K;++i)e.insert(i,TTT,1),e.insert(TTT,i,0);

    return Dinic()==C;
}

int main()
{
    int    i,j,k,l,r;

    scanf("%d%d%d",&K,&C,&M);
    n=K+C;
    for(i=1;i<=n;++i)
        for(j=1;j<=n;++j)
        {
            scanf("%d",&Map[i][j]);
            if(Map[i][j]==0)Map[i][j]=0x3f3f3f3f;
        }

    for(k=1;k<=n;++k)
        for(i=1;i<=n;++i)
            for(j=1;j<=n;++j)
                if(Map[i][k]+Map[k][j]<Map[i][j])
                    Map[i][j]=Map[i][k]+Map[k][j];

    l=1;r=0x7fffffff;SSS=++n,TTT=++n;
    while(l<r-1)
    {
        int    mid=l+((r-l)>>1);
        if(Check(mid))r=mid;
        else    l=mid;
    }

    printf("%d\n",r);

    return 0;
}