HDU 3549 Flow Problem 最大流 最小增广路 SAP算法 从EK算法的753MS降到了46MS

EK 算法的链接：http://blog.csdn.net/ipqhjjybj/article/details/9171181

SAP算法果然对这个优化了好多。。时间直接从753MS降到46MS呀。。

题目描述

Network flow is a well-known difficult problem for ACMers. Given a graph, your task is to find out the maximum flow for the weighted directed graph.

Input:

The first line of input contains an integer T, denoting the number of test cases.

For each test case, the first line contains two integers N and M, denoting the number of vertexes and edges in the graph. (2 <= N <= 15, 0 <= M <= 1000)

Next M lines, each line contains three integers X, Y and C, there is an edge from X to Y and the capacity of it is C. (1 <= X, Y <= N, 1 <= C <= 1000)

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>

#include <iostream>
#include <cmath>
#include <algorithm>
#include <numeric>
#include <utility>

#include <cstring>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <string>
using namespace std;

#define inf 0x3f3f3f3f
#define MAXN 20
#define MAXM 1005
#define clr(x,k) memset((x),(k),sizeof(x))
#define cpy(x,k) memcpy((x),(k),sizeof(x))
#define Base 10000

typedef vector<int> vi;
typedef stack<int> si;
typedef vector<string> vs;
#define sz(a) int((a).size())
#define pb push_back
#define all(c) (c).begin(),(c).end()
#define rep(i,n) for(int i = 0;i < n;++i)
#define foreach(it,c) for(vi::iterator it = (c).begin();it != (c).end();++it)

#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))

struct node{
int c,next,to;
}edges[MAXM];
int box[MAXN],n,m,tot;
void add(int a,int b,int c){
edges[tot].to=b;
edges[tot].c=c;
edges[tot].next=box[a];
box[a]=tot++;
}
int Map[MAXN][MAXN]; //map:邻接数组 因为可能有重边情况。优化下，再用SAP算法
void init(){
clr(Map,0);
for(int i = 0,a,b,c;i < m;i++){
scanf("%d %d %d",&a,&b,&c);
Map[a][b] += c;
}
tot=2;
clr(box,-1);
for(int i = 1;i <= n;i++)
for(int j = 1;j <= n;j++)
if(Map[i][j])
add(i,j,Map[i][j]),add(j,i,0);
}
int SAP_Max_Flow(int start,int end,int N){
int numh[MAXN],h[MAXN],curedges[MAXN],pre[MAXN];
//numh: 用于GAP优化的统计高度数量数组；h:距离标号数组;curedges;当前弧数组,pre,前驱数组
int cur_flow,flow_ans=0,u,tmp,neck,i;
clr(h,0);
clr(numh,0);
clr(pre,-1);
for(i = 1;i <= n;i++)
curedges[i]=box[i];
numh[0]=N;
u=start;
while(h[start]<N){
if(u==end){
cur_flow=inf;
for(i=start;i!=end;i=edges[curedges[i]].to){
if(cur_flow>edges[curedges[i]].c){
neck=i;
cur_flow=edges[curedges[i]].c;
}
}
for(i=start;i!=end;i=edges[curedges[i]].to){
tmp=curedges[i];
edges[tmp].c-=cur_flow;
edges[tmp^1].c+=cur_flow;
}
flow_ans+=cur_flow;
u=neck;
}
for(i=curedges[u];i!=-1;i=edges[i].next){
if(edges[i].c&&h[u]==h[edges[i].to]+1)
break;
}
if(i!=-1){
curedges[u]=i;
pre[edges[i].to]=u;
u=edges[i].to;
}
else{
if(0==--numh[h[u]])break;
curedges[u]=box[u];
for(tmp=N,i=box[u];i!=-1;i=edges[i].next)
if(edges[i].c)
tmp=min(tmp,h[edges[i].to]);
h[u]=tmp+1;
++numh[h[u]];
if(u!=start) u=pre[u];
}
}
return flow_ans;
}
int main(){
//freopen("3549.in","r",stdin);
int t,tt=0;
scanf("%d",&t);
while(t--){
scanf("%d %d",&n,&m);
init();
printf("Case %d: %d\n",++tt,SAP_Max_Flow(1,n,n));
}
return 0;
}