网络流学习(三)POJ2112

POJ2112,用Floyd+二分+最大流.我的最大流算法用的是Dinic算法,但是不知道为什么写出来就TLE了,..,..,..
后来发现被书上的代码误导了,书上把层次网络都全部求出来了,其实是没必要的,只要用一个数组对每个点的层次标记就好了.
附上Dinic算法的思路:
Dinic算法的思路也是分阶段地在层次网络中增广.它与最短增广路算法的不同之处是最短增广路算法每个阶段执行完一次BFS增广后,要重新启动BFS从源点Vs开始寻找另外一条增广路;而在Dinic算法中,只需一次DFS就可以实现多次增广.Dinic算法的具体步骤如下:
(1)初始化容量网络和网络流;
(2)构造残留网络和层次网络,若汇点不在层次网络中,则算法结束;
(3)在层次网络中用一次DFS过程进行增广,DFS执行完毕,该阶段的增广也执行完毕.
(4)转步骤(2).
下面是POJ2112的AC代码

#include <iostream>
#include <queue>
#include <cstdio>
#include <cstring>
using namespace std;
#define MAXN 300
#define inf 100000000
int dis[MAXN][MAXN];//连接任意点的最短路径
int Map[MAXN][MAXN];//容量网络
int sign[MAXN];//层次网络,并不直接得出,以层次标记获得
int K,C,M;

int min(int a,int b)
{
    return a<b?a:b;
}

void Build_Map(int Max_Dis)//构建容量网络
{
    int i,j;
    memset(Map,0,sizeof(Map));
    for(i=K+1;i<=K+C;i++) Map[0][i]=1;
    for(i=1;i<=K;i++) Map[i][K+C+1]=M;
    for(i=K+1;i<=K+C;i++)
    {
        for(j=1;j<=K;j++)
        {
            if(dis[i][j]<=Max_Dis)
               Map[i][j]=1;
        }
    }
}

bool BFS()
{
    queue<int> q;
    memset(sign,0,sizeof(sign));
    sign[0]=1;
    q.push(0);
    while(!q.empty())
    {
        int f=q.front();
        q.pop();
        for(int i=0;i<=K+C+1;i++)
        {
            if(!sign[i]&&Map[f][i])
            {
                sign[i]=sign[f]+1;
                q.push(i);
            }
        }
    }
    if(!sign[K+C+1]) return false;
    else return true;
}

int DFS(int v,int sum)
{
    int i,s,t;
    if(v==K+C+1) return sum;
    s=sum;
    for(i=0;sum&&i<=K+C+1;i++)
    {
        if(Map[v][i]&&sign[v]+1==sign[i])
        {
            t=DFS(i,min(Map[v][i],sum));
            Map[v][i]-=t;
            Map[i][v]+=t;
            sum-=t;
        }
    }
    return s-sum;
}

int main()
{
    int i,j,k,n;
    //Floyd求任意两点间距离
    scanf("%d%d%d",&K,&C,&M);
    n=K+C;
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            scanf("%d",&dis[i][j]);
            if(dis[i][j]==0)
                dis[i][j]=inf;
        }
    }
    for(k=1;k<=n;k++)
    {
        for(i=1;i<=n;i++)
        {
           if(dis[i][k]!=inf)
           {
            for(j=1;j<=n;j++)
            {
                dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
            }
           }
        }
    }
    int L=0,R=10000;
    while(L<R)
    {
        int mid=(L+R)/2;
        int ans=0;
        Build_Map(mid);
        while(BFS())
            ans+=DFS(0,inf);
        if(ans>=C) R=mid;
        else L=mid+1;
    }
    printf("%d\n",R);
    return 0;
}