学最大流,在网上和书上看了别人的代码,感觉不太好理解,有些乱。整理了一下,形成了自己的模板。
#define INF 0xffffff
#define MAX_NODE 205
#define MAX_EDGE 205
struct node{
int next,c,to;
node(){}
node(int c,int to,int next):c(c),to(to),next(next){}
}edge[MAX_EDGE];
int head[MAX_NODE],dis[MAX_NODE],gap[MAX_NODE],cur[MAX_NODE],pre[MAX_NODE];
int top;
int n,m,START,END;
void add(int a,int b,int c){
edge[top]=node(c,b,head[a]);
head[a]=top++;
edge[top]=node(0,a,head[b]);
head[b]=top++;
}
int SAP(){
int max_flow=0,flow=INF,u=START,v;
for(int i=1;i<=n;++i)dis[i]=gap[i]=0,cur[i]=head[i];
gap[0]=n;pre[START]=START;
while(dis[START]<n){
for(int &i=cur[u];i!=-1;i=edge[i].next){
v=edge[i].to;
if(edge[i].c&&dis[u]==dis[v]+1){
break;
}
}
if(cur[u]!=-1){
flow=min(flow,edge[cur[u]].c);
pre[v]=u;u=v;
if(v==END){
max_flow+=flow;
int i=u;
while(i!=START){
i=pre[i];
edge[cur[i]].c-=flow;
edge[cur[i]^1].c+=flow;
}
flow=INF;
u=START;
}
}
else{
int minds=n;
for(int i=head[u];i!=-1;i=edge[i].next){
v=edge[i].to;
if(edge[i].c&&minds>dis[v]){
minds=dis[v];
cur[u]=i;
}
}
if(--gap[dis[u]]==0)break;
++gap[dis[u]=minds+1];
u=pre[u];
}
}
return max_flow;
}