dinic 模板

自己整理,仅做模板保存。

cur【】数组是当前弧优化的辅助数组。
当前弧优化是dinic算法中十分重要的优化。cur【】数组的使用方法即首先将head【】数组复制一遍。在dfs中再改变当前榨取到哪一条边。每次榨取,该边所有流量必定被榨干,下一次搜索,直接从这条边开始就好。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=105;
const int maxe=10010;
const ll inf=1e15;

int dep[maxn];
int head[maxn];
int cur[maxn];  //当前弧优化的辅助数组
int step;

struct ED
{
    int to,next;
    ll val;
}edge[maxe];

void add(int t,ll v,int x)
{
    edge[step].to=t;
    edge[step].val=v;
    edge[step].next=head[x];
    head[x]=step;
    step++;
}

bool bfs(int s,int t)
{
    memset(dep,0,sizeof(dep));
    queue<int>Q;
    Q.push(s);
    dep[s]=1;
    cur[s]=head[s];
    while(!Q.empty())
    {
        s=Q.front();
        Q.pop();
        for(int i=head[s];i!=-1;i=edge[i].next)
        {
            int y=edge[i].to;
            if(edge[i].val>0&&dep[y]==0)
            {
                dep[y]=dep[s]+1;
                cur[y]=head[y];
                if(y==t) return true;
                Q.push(y);
            }
        }
    }
    return false;
}

ll dfs(int s,ll flow,int t)
{
    if(s==t||flow<=0) return flow;
    ll rest=flow;
    for(int i=cur[s];i!=-1;i=edge[i].next)
    {
        cur[s]=i;  //弧优化记录当前“榨取”到哪条边了
        int y=edge[i].to;
        if(edge[i].val>0&&dep[y]==dep[s]+1)
        {
            ll temp=dfs(y,min(rest,edge[i].val),t);
            if(temp<=0) dep[y]=0;  //该点无法再提供流量,删掉
            rest-=temp;
            edge[i].val-=temp;
            edge[i^1].val+=temp;
            if(rest<=0) break;
        }
    }
    return flow-rest;
}

ll dinic(int s,int t)
{
    ll ans=0;
    while(bfs(s,t))
        ans+=dfs(s,inf,t);
    return ans;
}

int main()
{
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int n,m,s,t;
    cin>>n>>m>>s>>t;
    step=2;  //从偶数开始记录边,方便记录反向边
    memset(head,-1,sizeof(head));
    for(int i=1;i<=m;i++)
    {
        int x,y;
        ll f;
        cin>>x>>y>>f;
        add(y,f,x);
        add(x,0,y);
    }
    ll ans=dinic(s,t);
    cout<<ans<<endl;
    return 0;
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值