最近在学习网络流部分,之前看了一下dinic算法,一直没机会尝试一下,今天终于体验了一下,然后就做一下总结
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4280
在做题过程中使用了邻接表,dinic算法,以及一个超神的扩展栈空间的语句
具体步骤在下面的代码中体现
#include<stdio.h>
#include<string.h>
#pragma comment(linker, "/STACK:1024000000,1024000000") //用来提高栈空间,新技能get!
const int MAXN=100005;
const int inf=0x0fffffff;
int m,n,top,head[MAXN],s,e,mw,me,level[MAXN],Queue[MAXN],top_Q;
struct Edge
{
int to,c,next;
}edge[2*MAXN];
int getMin(int a,int b)
{
return a>b?b:a;
}
void addEdge(int a,int b,int c)
{
edge[top].to=b;
edge[top].c=c;
edge[top].next=head[a];
head[a]=top++;
}
bool bfs()
{
memset(level,-1,sizeof(level));
top_Q=0;
level[s]=0;
Queue[top_Q++]=s;
for(int i=0;i<top_Q;i++)
{
for(int j=head[Queue[i]];j!=-1;j=edge[j].next)
{
if(level[edge[j].to]==-1&&edge[j].c>0)
{
level[edge[j].to]=level[Queue[i]]+1;
Queue[top_Q++]=edge[j].to;
if(edge[j].to==e)//只要bfs到e即可,后面的在dfs中用不上,不跳出会超时
return true;
}
}
}
return false;
}
int dfs(int p,int a)//核心部分:p是当前点,a是能流入到该点的最大流量
{
if(p==e||a==0)
return a;
int temp,flow=0;
for(int i=head[p];i!=-1;i=edge[i].next)
{
if(edge[i].c>0&&level[edge[i].to]==level[p]+1)
{
temp=getMin(a,edge[i].c);//a代表剩余的最大流量
temp=dfs(edge[i].to,temp);//找到一条增广路不立即跳出,相当于寻找分流量
if(temp>0)
{
edge[i].c-=temp;//直接处理,不需要寻找路径处理
edge[i^1].c+=temp;
flow+=temp;//flow用来记录能达到的最大分流量之和
a-=temp;
if(a==0)//表示已经达到最大流量了,即使还能向下找到分流量也没用,所以要跳出
break;
}
}
}
if(flow==0)
level[p]=0;
return flow;//返回向下最大流量
}
void dinic()
{
int ans=0;
while(bfs())//dinic就是先bfs,再dfs
{
int dx;
if(dx=dfs(s,inf))
ans+=dx;
}
printf("%d\n",ans);
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
mw=inf;
me=-inf;
for(int i=1;i<=n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
if(x>me)
{
me=x;
e=i;
}
if(x<mw)
{
mw=x;
s=i;
}
}
top=0;
memset(head,-1,sizeof(head));
for(int i=0;i<m;i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
addEdge(a,b,c);
addEdge(b,a,c);
}
dinic();
}
return 0;
}