最大流EK算法模板

有关最大流的解释看《算法导论》和博客中的相关内容

模板(C++):
#include <iostream>
#include <queue>
//#include <conio.h>
using namespace std;
#define arraysize 201
int maxData = 0x7fffffff;
int capacity[arraysize][arraysize]; //记录残留网络的容量
int flow[arraysize];                //标记从源点到当前节点实际还剩多少流量可用
int pre[arraysize];                 //标记在这条路径上当前节点的前驱,同时标记该节点是否在队列中
int n,m;
queue<int> myqueue;
int BFS(int src,int des)            //EK算法使用BFS寻找增广路径
{
    int i,j;
    while(!myqueue.empty())         //队列清空
        myqueue.pop();
    for(i=1;i<m+1;++i)              //初始化前驱
    {
        pre[i]=-1;
    }
    pre[src]=0;
    flow[src]= maxData;             //初始化源点的流量为无穷大
    myqueue.push(src);
    while(!myqueue.empty())
    {
        int index = myqueue.front();
        myqueue.pop();
        if(index == des)            //找到了增广路径
            break;
        for(i=1;i<m+1;++i)          //遍历所有的结点
        {
            if(i!=src && capacity[index][i]>0 && pre[i]==-1)
            {
                 pre[i] = index;              //记录前驱
                 flow[i] = min(capacity[index][i],flow[index]);   //关键:迭代的找到增量
                 myqueue.push(i);              
            }
        }
    }
    if(pre[des]==-1)      //残留图中不再存在增广路径 
        return -1;
    else
        return flow[des];
}
int maxFlow(int src,int des)
{
    int increasement= 0;
    int sumflow = 0;
    while((increasement=BFS(src,des))!=-1)
    {
         int k = des;          //利用前驱寻找路径
         while(k!=src)
         {
              int last = pre[k];
              capacity[last][k] -= increasement; //改变正向边的容量
              capacity[k][last] += increasement; //改变反向边的容量
              k = last;
         }
         sumflow += increasement;
    }   
    return sumflow;
}


 

posted @ 2010-05-25 21:38  北海小龙  阅读(568)  评论(0编辑  收藏  举报