最大流dinic算法的优化模板 当前弧优化,多路增广

#include<string.h>
#include<cstdio>
#include<algorithm>
#include<limits.h>
#include<queue>
using namespace std;
const int M=420;
const int inf=0x3f3f3f3f;
struct aa
{
    int u,v,r,next;
} edge[888888];
int cnt,dis[M],s,f,n,m,cost[M],pre[M],head[M];
void add(int u,int v,int r)
{
    edge[cnt].v=v;
    edge[cnt].u=u;
    edge[cnt].r=r;
    edge[cnt].next=pre[u];
    pre[u]=cnt++;
    edge[cnt].v=u;
    edge[cnt].u=v;
    edge[cnt].r=0;
    edge[cnt].next=pre[v];
    pre[v]=cnt++;
}
bool bfs()//建立层次图
{

    memset(dis,0,sizeof(dis));
    dis[s]=1;
    queue<int>que;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        for(int i=pre[u]; i!=-1; i=edge[i].next)
        {
            int v=edge[i].v;
            int r=edge[i].r;
            if(dis[v]==0&&r>0)
            {
                dis[v]=dis[u]+1;
                que.push(v);
            }
        }
    }
    return dis[f]>=1;
}
int dfs(int now,int Min)//多路增广
{
    int a;
    if(now==f)
        return Min;
    int cost=0;
    for(int &i=head[now]; i!=-1; i=edge[i].next)
    {
        int v=edge[i].v;
        int r=edge[i].r;
        if(dis[v]==dis[now]+1&&r>0 &&(a=dfs(v,min(Min-cost,r))))
        {
            edge[i].r-=a;
            edge[i^1].r+=a;
            cost+=a;//汇聚各个方向的流量
            if(Min==cost)//当某几个方向的流量相加等于当前最大流量的时候就无法对别的路进行增广了
                break;
        }
    }
    return cost;//返回当前汇聚的流量
}
int Max_f()
{
    int sum=0;
    while(bfs())
    {
        for(int i=0; i<=2*n; i++)//当前弧优化
            head[i]=pre[i];
        while(int temp=dfs(s,inf))//一次bfs多次增广
            sum+=temp;
    }
    return sum;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        scanf("%d%d",&s,&f);
        memset(pre,-1,sizeof(pre));
        f=f+n;
        cnt=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&cost[i]);
            add(i,n+i,cost[i]);
        }
        while(m--)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            add(n+u,v,inf);
            add(v+n,u,inf);
        }
        int ans=Max_f();
        printf("%d\n",ans);
    }
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值