0
点赞
收藏
分享

微信扫一扫

EK算法求最大流问题

ivy吖 2022-02-16 阅读 44

上一道裸题吧!

EK算法是基于增广路算法进行广搜。增广路是指在残差网络上的一条路径,每条边的权重都大于0。我们有如下的结论:如果残差网络中存在增广路,流量就可以增大(增广路定理)。所以我们从起始位置开始去搜索一条增广路径,找到后对残差网络进行更新,直到找不到增广路,那么此时得到的就是最大流的网络。 

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <queue>
#define INF 0x3f3f3f3f

using namespace std;

int mp[233][233];
int flow[233], pre[233];
int n;

int bfs()
{
    queue <int> q;
    memset(pre, -1, sizeof(pre));
    q.push(1);
    flow[1] = INF;
    pre[1] = 0;
    int u, v, i;
    while (!q.empty())
    {
        u = q.front();
        q.pop();
        if (u == n) return flow[u];
        for (i = 1; i <= n; ++i)
        {
            if (pre[i] == -1 && mp[u][i] > 0)
            {
                flow[i] = min(mp[u][i], flow[u]);
                pre[i] = u;
                q.push(i);
            }
        }
    }
    if (pre[n] = -1) return 0;
    return flow[n];
}

int max_flow()
{
    int f, sumflow = 0;
    while ((f = bfs()) != 0)
    {
        int k = n;
        while (k != 1)
        {
            int last = pre[k];
            mp[last][k] -= f;
            mp[k][last] += f;
            k = last;
        }
        sumflow += f;
    }
    return sumflow;
}

int main()
{
    int m, u, v, w;
    while (~scanf("%d %d", &m, &n))
    {
        memset(mp, 0, sizeof(mp));
        while (m--)
        {
            scanf("%d %d %d", &u, &v, &w);
            if (u == v) continue;
            mp[u][v] += w;
        }
        printf("%d\n", max_flow());
    }
    return 0;
}
举报

相关推荐

0 条评论