题目:USACO Training 4.2(在官网上提交需加文件输入输出)、洛谷P2740、codevs1993、POJ1273(多数据读入)、HDU1532(多数据读入)。
题目大意:有n条有向边连接着m个点,每条边有一个最大容量,求从1到n最多流多少水。
解题思路:最大流模板题,我A的第一道网络流题目。以下是Dinic算法的代码。
C++ Code:
#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 1000000000
int n,m,iter[202],level[202];
queue<int>q;
struct edges{
int to,cap,rev;
edges(int t,int c,int r):to(t),cap(c),rev(r){}
};
vector<edges>G[202];
inline void addedge(int u,int v,int dis){
G[u].push_back(edges(v,dis,G[v].size()));
G[v].push_back(edges(u,0,G[u].size()-1));
}
void bfs(int s){
memset(level,-1,sizeof(level));
level[s]=0;
q.push(s);
while(!q.empty()){
int u=q.front();q.pop();
for(int i=0;i<G[u].size();++i){
edges& e=G[u][i];
if(level[e.to]<0&&e.cap>0){
level[e.to]=level[u]+1;
q.push(e.to);
}
}
}
}
int dfs(int u,int t,int f){
if(u==t)return f;
for(int& i=iter[u];i<G[u].size();++i){
edges& e=G[u][i];
if(e.cap>0&&level[u]<level[e.to]){
int d=dfs(e.to,t,min(f,e.cap));
if(d){
e.cap-=d;
G[e.to][e.rev].cap+=d;
return d;
}
}
}
return 0;
}
int max_flow(int s,int t){
int flow=0;
while(1){
bfs(s);
if(level[t]<0)break;
memset(iter,0,sizeof(iter));
int f;
while((f=dfs(s,t,INF))>0)flow+=f;
}
return flow;
}
int main(){
scanf("%d%d",&m,&n);
for(int i=1;i<=m;++i){
int x,y,cap;
scanf("%d%d%d",&x,&y,&cap);
addedge(x,y,cap);
}
printf("%d\n",max_flow(1,n));
return 0;
}
以下是POJ和HDU的多数据版本,注意初始化即可。
C++ Code:
#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
#define INF 1000000000
int n,m,iter[202],level[202];
queue<int>q;
struct edges{
int to,cap,rev;
edges(int t,int c,int r):to(t),cap(c),rev(r){}
};
vector<edges>G[202];
inline void addedge(int u,int v,int dis){
G[u].push_back(edges(v,dis,G[v].size()));
G[v].push_back(edges(u,0,G[u].size()-1));
}
void bfs(int s){
memset(level,-1,sizeof(level));
level[s]=0;
q.push(s);
while(!q.empty()){
int u=q.front();q.pop();
for(int i=0;i<G[u].size();++i){
edges& e=G[u][i];
if(level[e.to]<0&&e.cap>0){
level[e.to]=level[u]+1;
q.push(e.to);
}
}
}
}
int dfs(int u,int t,int f){
if(u==t)return f;
for(int& i=iter[u];i<G[u].size();++i){
edges& e=G[u][i];
if(e.cap>0&&level[u]<level[e.to]){
int d=dfs(e.to,t,min(f,e.cap));
if(d){
e.cap-=d;
G[e.to][e.rev].cap+=d;
return d;
}
}
}
return 0;
}
int max_flow(int s,int t){
int flow=0;
while(1){
bfs(s);
if(level[t]<0)break;
memset(iter,0,sizeof(iter));
int f;
while((f=dfs(s,t,INF))>0)flow+=f;
}
return flow;
}
int main(){
while(scanf("%d%d",&m,&n)!=EOF){
for(int i=1;i<=201;i++)G[i].clear();
for(int i=1;i<=m;++i){
int x,y,cap;
scanf("%d%d%d",&x,&y,&cap);
addedge(x,y,cap);
}
printf("%d\n",max_flow(1,n));
}
return 0;
}