hdu 1532 网络流EK模版

/*
题意：给出排水沟的图，以及沟渠流量，求最大流出速率

题解：最大流，EK算法，完全模版；
*/
#include <iostream>
#include <cstring>
#include <queue>

#define EMAX 405
#define VMAX 205

using namespace std;

int EN;//边的总数

int head[VMAX];//用邻接表表示图
int pre[VMAX];//记录路径
int path[EMAX];//记录该点边的位置
struct edge
{
int to;
int weight;//流量
int next;
}e[EMAX];

void insert(int u, int v, int w)
{
e[EN].next = head[u];
e[EN].to = v;
e[EN].weight = w;
head[u] = EN++;
e[EN].next = head[v];
e[EN].to = u;
e[EN].weight = 0;//需要插入反向边
head[v] = EN++;
}

int flow_ek(int s, int t)
{
queue<int> Q;
int ret = 0;
while (true)
{
memset(pre,-1,sizeof(pre));
while (!Q.empty())
Q.pop();
Q.push(s);
while (!Q.empty())//BFS
{
int u = Q.front();
Q.pop();
for(int i=head[u]; i!=-1; i=e[i].next)
{
if (pre[e[i].to]==-1 && e[i].weight>0)
{
pre[e[i].to] = u;//记录路径
path[e[i].to] = i;//记录该点边的所在位置，方便后面直接查询
Q.push(e[i].to);
}
}
if (pre[t] != -1)//终点的pre不为-1,表示找到一条增广路径
break;
}
if (pre[t] == -1)//当BFS后找不到增广路径则结束循环
break;
int mw = -1;
for(int v=t; v!=s; v=pre[v])//找出当前路径中的流量的最小容量
{
if (mw == -1 || mw > e[path[v]].weight)
mw = e[path[v]].weight;
}
for(int v=t; v!=s; v=pre[v])//修改路径的容量，求出残余网络
{
e[path[v]].weight -= mw;
e[path[v]^1].weight += mw;//更新逆向边
}
ret += mw;
}
return ret;
}

int main(void)
{
int n,m,a,b,c;
while (cin >> n >> m)
{
memset(head,-1,sizeof(head));
EN = 0;
for(int i=0; i<n; i++)
{
cin >> a >> b >> c;
insert(a,b,c);
}
cout << flow_ek(1,m) << endl;
}
return 0;
}