(网络流 模板 Dinic) Drainage Ditches --POJ --1273

链接：

http://poj.org/problem?id=1273

代码：

//Dinic

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;

const int MAXN = ;
const int oo = 1e9+;

struct Edge
{
    int v, flow, next;
}edge[MAXN];
int Head[MAXN], cnt;
int Layer[MAXN];

void InIt()
{
    cnt = ;
    memset(Head, -, sizeof(Head));
}

void AddEdge(int u, int v, int flow)
{
    edge[cnt].v = v;
    edge[cnt].flow = flow;
    edge[cnt].next = Head[u];
    Head[u] = cnt++;

    swap(u, v);

    edge[cnt].v = v;
    edge[cnt].flow = ;
    edge[cnt].next = Head[u];
    Head[u] = cnt++;

}

bool BFS(int start, int End)   /// 分层
{
    memset(Layer, , sizeof(Layer));
    queue<int> Q;
    Q.push(start);
    Layer[start] = ;

    while(Q.size())
    {
        int u = Q.front();Q.pop();

        if(u == End)return true;

        for(int j=Head[u]; j!=-; j=edge[j].next)
        {
            int v = edge[j].v;

            if(Layer[v] ==  && edge[j].flow) /// 判断Layer[v]==0，可以减少循环
            {
                Layer[v] = Layer[u] + ;
                Q.push(v);
            }
        }
    }

    return false;
}

int DFS(int u, int MaxFlow, int End)  /// DFS返回的是这条路上的最大流（最小值）
{                                     ///MaxFlow代表从源点起的最大流量
    if(u == End)return MaxFlow;

    int uflow = ;                    ///从u点到汇点的最大流量

    for(int j=Head[u]; j!=-; j=edge[j].next)
    {
        int v = edge[j].v;

        if(Layer[v]==Layer[u]+ && edge[j].flow)
        {
            int flow = min(edge[j].flow, MaxFlow-uflow);
            flow = DFS(v, flow, End);

            edge[j].flow -= flow;
            edge[j^].flow += flow;
            uflow += flow;

            if(uflow == MaxFlow)
                break;
        }
    }

    if(uflow == )
        Layer[u] = ;

    return uflow;
}

int Dinic(int start, int End)
{
    int MaxFlow = ;

    while(BFS(start, End) == true)  /// 先找出一条从Start到End的路
        MaxFlow += DFS(start, oo, End); 

    return MaxFlow;
}

int main()
{
    int N, M;

    while(scanf("%d%d", &M, &N) != EOF)
    {
        int u, v, flow;

        InIt();

        while(M--)
        {
            scanf("%d%d%d", &u, &v, &flow);
            AddEdge(u, v, flow);
        }

        printf("%d\n", Dinic(, N));
    }

    return ;
}