LightOJ - 1153-Internet Bandwidth（最大流）

原题地址：点击打开链接

 该题是个简单的最大流模板题，唯一需要注意的是无向图，需要双向建边。

#include<stdio.h>
#include<string.h>
#include<vector>
#define INF 1<<31-1
#define min(x,y)(x<y?x:y)
using namespace std;
struct Edge
{
	int to;
	int cap;
	int rev;
};
vector<Edge>g[110];
int used[110]; 
void add_edge(int from,int to,int cap)
{
	g[from].push_back((Edge){to,cap,g[to].size()});
	g[to].push_back((Edge){from,0,g[from].size()-1});
}
int dfs(int u,int t,int f)
{
	if(u==t)
	  return f;
	  used[u]=1;
	for(int i=0;i<g[u].size();i++)
	{
		Edge &e=g[u][i];
		int v=e.to;
		if(!used[v]&&e.cap>0)
		{
			int d=dfs(v,t,min(e.cap,f));
			if(d>0)
			{
				e.cap-=d;
				g[e.to][e.rev].cap+=d;
				return d;
			}
		}
	}
	return 0;
}
int max_flow(int s,int t)
{
	int res=0,f;
	while(1)
	{
		f=INF;
		memset(used,0,sizeof(used));
		int d=dfs(s,t,f);
		if(d<=0)
			break;
		res+=d;
	}
	return res;
}
int main()
{
	int i,c,n,s,t,m,from,to,cap,k=0;
	scanf("%d",&c);
	while(c--)
	{
		scanf("%d",&n);
		scanf("%d%d%d",&s,&t,&m);
		for(i=1;i<=n;i++)
			g[i].clear();
		for(i=0;i<m;i++)
		{
			scanf("%d%d%d",&from,&to,&cap);
			add_edge(from,to,cap);
			add_edge(to,from,cap);
		}
		int res=max_flow(s,t);
		printf("Case %d: %d\n",++k,res);
	}
	return 0;
} 
/*


2
4
1 4 5
1 2 20
1 3 10
2 3 5
2 4 10
3 4 20
4
1 4 2
1 4 20
1 4 20
*/