草地排水

最大流模板题 

#include 
#include 
#include 
#include 
using namespace std;
int read()
{
	int x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9')
	{
		if(c=='-')  f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')
	{
		x=x*10+c-'0';
		c=getchar();
	}
	return f*x;
}
const int N=300,M=300,MAX=2147483647;
int n,m,a1,b1,c1,s,t,dis[N],cur[N],ans;
int had[N],to[M*2],cap[M*2],nxt[M*2],p=1;
void add(int x,int y,int z)
{
	nxt[++p]=had[x];
	to[p]=y;
	cap[p]=z;
	had[x]=p;
}
int bfs(int s,int t)
{
	queue  que;
	memset(dis,0,sizeof(dis));
	dis[s]=1;  que.push(s);
	while(que.size())
	{
		int u=que.front();
		que.pop();  cur[u]=had[u];
		for(int i=had[u];i;i=nxt[i])
		{
			int v=to[i];
			if(!cap[i]||dis[v])  continue;
			dis[v]=dis[u]+1;
			que.push(v);
		}
	}
	return dis[t];
}
int dfs(int u,int lim)
{
	if(u==t||!lim)  return lim;
	int flow=0;
	for(int i=cur[u];i;i=nxt[i])
	{
		int v=to[i];
		cur[u]=i;
		if(cap[i]&&dis[v]==dis[u]+1)
		{
			int x=dfs(v,min(cap[i],lim-flow));
			flow+=x;
			cap[i]-=x;
			cap[i^1]+=x;
			if(lim==flow)  break;
		}
	}
	return flow;
}
int main()
{
	m=read();  n=read();
	s=1;  t=n;
	for(int i=1;i<=m;i++)
	{
		a1=read();  b1=read();  c1=read();
		add(a1,b1,c1);  add(b1,a1,0);
	}
	while(bfs(s,t))  ans+=dfs(s,MAX);
	cout<

 

你可能感兴趣的:(草地排水)