網絡流之最大流EK --- poj 1459

題目連接

 

本篇博客延續上篇博客（最大流Dinic算法）的內容，這次使用EK算法解決最大流問題。

EK算法思想：在圖中搜索一條從源點到匯點的擴展路，須要記錄這條路徑，將這條路徑的最大可行流量 liu 增長到結果ans中，而後反向從匯點到源點更新這條路徑上的每條邊的權值（減去這次的liu），同時反向邊的權值也須要更新（加上這次的liu）。而後再搜索新的擴展路……，循環，直到找不到新的擴展路，此時的ans就是最大流了。

注：EK算法解決最大流時，我看別人都是使用矩陣創建的圖，這樣反向更新擴展路徑上的邊權時，只須要以前記錄每一個點的父親節點便可。我是在前一篇的Dinic的前向星代碼上修改成EK算法，由父親和孩子節點編號不能直接得出鏈接的邊號，因此須要另外用一個變量 edgeIndex[v] 記錄從 u 到 v 的邊號，這樣方便反向修改邊權值。

 

代碼以下：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <queue>
using namespace std;
const int N = 105;
const int MAXN = 0x3fffffff;

struct Edge {
    int to;
    int value;
    int next;
}e[2 * N*N];
int head[N], cnt;
int p[N], fa[N], edgeIndex[N];
int n, np, nc, m;

void insert(int u, int v, int value) {
    e[++cnt].to = v;
    e[cnt].value = value;
    e[cnt].next = head[u];
    head[u] = cnt;
}

void init() {
    memset(head, -1, sizeof(head));
    cnt = -1;
}

int BFS() {
    int ans = 0;
    while (1)
    {
        memset(p, -1, sizeof(p));
        queue<int> Q;
        Q.push(0);
        p[0] = MAXN;
        while (!Q.empty()) {
            int u = Q.front();
            Q.pop();
            for (int edge = head[u]; edge != -1; edge = e[edge].next) {
                int v = e[edge].to;
                if (p[v] ==-1 && e[edge].value > 0) {
                    p[v] = min(p[u], e[edge].value);
                    fa[v] = u;
                    edgeIndex[v] = edge;
                    Q.push(v);
                    if (v == n + 1) goto endw;
                }
            }
        }
    endw:;
        if (p[n+1] == -1) return ans;
        else {
            ans += p[n + 1];
            int x = n + 1;
            while (x) {
                int edge = edgeIndex[x];
                e[edge].value -= p[n + 1];
                e[edge ^ 1].value += p[n + 1];
                x = fa[x];
            }
        }
    }
}



int main()
{
    while (scanf("%d%d%d%d", &n, &np, &nc, &m) != EOF) {
        init();
        int u, v, z;
        for (int i = 0; i < m; i++) {
            scanf(" (%d,%d)%d", &u, &v, &z);
            insert(u + 1, v + 1, z);
            insert(v + 1, u + 1, 0);
        }
        for (int i = 0; i < np; i++) {
            scanf(" (%d)%d", &u, &z);
            insert(0, u + 1, z);
            insert(u + 1, 0, 0);
        }
        for (int i = 0; i < nc; i++) {
            scanf(" (%d)%d", &u, &z);
            insert(u + 1, n + 1, z);
            insert(n + 1, u + 1, 0);
        }
        printf("%d\n", BFS());
    }
}