POJ-3204-Ikki's Story I - Road Reconstruction

最新推荐文章于 2017-03-31 18:34:31 发布

南宮逸辰

最新推荐文章于 2017-03-31 18:34:31 发布

阅读量602

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM POJ 文章标签：网络流 Dinic

本文链接：https://blog.youkuaiyun.com/z309241990/article/details/9755233

ACM 同时被 2 个专栏收录

716 篇文章

订阅专栏

POJ

378 篇文章

订阅专栏

本文介绍了一种解决最大流问题的方法——Dinic算法，并提供了详细的C++实现代码。文章通过具体实例展示了如何利用Dinic算法计算图的最大流，并通过DFS搜索确定哪些边的容量增加可以进一步提高最大流。

题目要求算出通过增加边的容量能够使得最大流增加的边数

思路：先用Dinic算出图的最大流，然后用DFS分别从始点和汇点进行搜索标记，最后枚举算出有多少点可以改变最大流即可

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=501;
const int maxm=200000;
const int inf=1<<29;
int n,m,e,st,des,head[maxn],pnt[maxm],nxt[maxm],cost[maxm],level[maxn],q[maxn];
bool visa[maxn],visb[maxn];
void AddEdge(int u,int v,int c)
{
    pnt[e]=v;nxt[e]=head[u];cost[e]=c;head[u]=e++;
    pnt[e]=u;nxt[e]=head[v];cost[e]=0;head[v]=e++;
}
bool BFS()
{
    memset(level,0,sizeof(level));
    int pre=0,last=1;
    q[pre]=st;
    level[st]=1;
    while(pre<last)
    {
	if(q[pre]==des)
	    return true;
	for(int i=head[q[pre]];i!=-1;i=nxt[i])
	    if(cost[i]&&!level[pnt[i]])
	    {
		level[pnt[i]]=level[q[pre]]+1;
		q[last++]=pnt[i];
	    }
	pre++;
    }
    return false;
}
int DFS(int u,int sum)
{
    if(u==des)
	return sum;
    for(int i=head[u],t;i!=-1;i=nxt[i])
	if(cost[i]&&level[pnt[i]]==level[u]+1&&(t=DFS(pnt[i],min(sum,cost[i]))))
	{
	    cost[i]-=t;
	    cost[i^1]+=t;
	    return t;
	}
    return level[u]=0;
}
void DFSA(int u)
{
    visa[u]=1;
    for(int i=head[u];i!=-1;i=nxt[i])
	if(!visa[pnt[i]]&&cost[i])
	    DFSA(pnt[i]);
}
void DFSB(int u)
{
    visb[u]=1;
    for(int i=head[u];i!=-1;i=nxt[i])
	if(!visb[pnt[i]]&&cost[i^1])
	    DFSB(pnt[i]);
}
void Dinic()
{
    while(BFS())
	while(1)
	{
	    int val=DFS(st,inf);
	    if(!val)
		break;
	}
    memset(visa,0,sizeof(visa));
    memset(visb,0,sizeof(visb));
    DFSA(st);
    DFSB(des);
    int ans=0;
    for(int i=0;i<e;i+=2)
    {
	if(!cost[i]&&visa[pnt[i^1]]&&visb[pnt[i]])
	    ans++;
    }
    printf("%d\n",ans);
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
	st=e=0,des=n-1;
	memset(head,-1,sizeof(head));
	for(int i=0;i<m;i++)
	{
	    int u,v,c;
	    scanf("%d%d%d",&u,&v,&c);
	    AddEdge(u,v,c);
	}
	Dinic();
    }
    return 0;
}