洛谷 3376_[模板]网络最大流

发布于 2019-05-18  989 次阅读


题目描述

如题,给出一个网络图,以及其源点和汇点,求出其网络最大流。

#include <stdio.h>
#include <string.h>
#include <queue>
#define maxn 400000
#define min(x,y) x<y?x:y
#define INF 0x7f7f7f7f
#define fill(x,i) memset(x,i,sizeof(x))
using namespace std;
struct edge{int y,w,rev,next;}e[maxn];
int ls[maxn],n,m,maxE=0,S,E,state[maxn];
bool vis[maxn];
int add(int x,int y,int w)
{
    e[++maxE]=(edge){y,w,maxE+1,ls[x]};
    ls[x]=maxE;
    e[++maxE]=(edge){x,0,maxE-1,ls[y]};
    ls[y]=maxE;
}
int bfs(int x)
{
    queue <int> t;
    t.push(x);
    fill(state,63);
    state[x]=0;
    while (!t.empty())
    {
        int tt=t.front();t.pop();
        for (int i=ls[tt];i;i=e[i].next)
        {
            if (e[i].w>0&&state[tt]+1<state[e[i].y])
            {
                state[e[i].y]=state[tt]+1;
                t.push(e[i].y);
                if (e[i].y==E)
                    return true;
            }
        }
    }
    return false;
}
int find(int x,int mn)
{
    if (x==E) return mn;
    for (int i=ls[x];i;i=e[i].next)
        if (state[x]+1==state[e[i].y]&&e[i].w>0)
        {
            int d=find(e[i].y,min(mn,e[i].w));
            if (d>0)
            {
                e[i].w-=d;
                e[e[i].rev].w+=d;
                return d;
            }
        }
    return 0;
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&S,&E);
    for (int i=1;i<=m;i++)
    {
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);
    }
    int ans=0;
    while (bfs(S))
    {
        ans+=find(S,INF);
    }
    printf("%d\n",ans);
}