POJ 1273 Drainage Ditches（dinic模板）

题目大意：

网络流的模板题。

解题思路：

模板。

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<set>
#define LL long long
#define db double
#define maxn 10000000
#define EPS 1e-15
#define max(a,b) ((a>b)?(a):(b))
#define min(a,b) ((a<b)?(a):(b))
using namespace std;

const int oo=1e9; // 无穷大
const int mm=111111; // 边的最大数量，原图两倍，双向。
const int mn=999; // 点的最大数量
int node,src,dest,edge;
// node节点数，src源点，dest汇点，edge统计边数
int ver[mm],flow[mm],next[mm];
// ver边指向的节点，flow容量，next链表下一条边
int head[mn],work[mn],dis[mn],q[mn];
//head节点链表头，work临时链表头，dis距离用来分层

// 初始化链表和图
void prepare(int _node,int _src,int _dest){
	node=_node, src=_src, dest=_dest;
	for (int i=0;i<node;++i) head[i]=-1;
	edge=0;
}
// 增加一条从u到v容量为c的边
void add(int u,int v,int c){
	ver[edge]=v, flow[edge]=c;
	next[edge]=head[u], head[u]=edge++;

	ver[edge]=u, flow[edge]=0;
	next[edge]=head[v], head[v]=edge++;
}
// 按照每个点和源点的最短距离分层
// 如果不能到达汇点说明算法结束
bool dinic_bfs(){
	int i,u,v,l,r=0;
	for (int i=0;i<node;++i) dis[i]=-1;
	dis[q[r++]=src]=0;
	for (l=0;l<r;l++)
		for (i=head[u=q[l]]; i>=0; i=next[i])
		if (flow[i] && dis[v=ver[i]]<0){
			dis[q[r++]=v]=dis[u]+1;
			if (v==dest) return 1;
		}
	return 0;
}
//寻找可行流的增光路算法，按节点的距离来找
int dinic_dfs(int u,int exp){
	if (u==dest) return exp;
	// work是临时链表头，这里用i引用它。
	//这样寻找过的边不再寻找
	for (int &i=work[u],v,tmp;i>=0;i=next[i])
		if (flow[i]
			&& dis[v=ver[i]]==dis[u]+1
			&& (tmp=dinic_dfs(v,min(flow[i],exp)))>0){
				flow[i]-=tmp;
				flow[i+1]+=tmp;
				//正反向边容量改变
				return tmp;
			}
	return 0;
}
// 求最大流，直到没有可行流
int dinic_flow(){
	int i,ret=0,delta;
	while (dinic_bfs()){
		for (i=0;i<node;++i)  work[i]=head[i];
		while (delta=dinic_dfs(src,oo)) ret+=delta;
	}
	return ret;
}
int main(){
	int n,m,u,v,c;
	while (scanf("%d%d",&n,&m)!=EOF){
		prepare(m+1,1,m);
		while (n--){
			scanf("%d%d%d",&u,&v,&c);
			add(u,v,c);
		}
		printf("%d\n",dinic_flow());
	}
	return 0;
}