1. 程式人生 > >Dinic最大流模板

Dinic最大流模板

struct Edge
{
    int from,to,cap,flow;
    Edge(int u,int v,int c,int f):from(u),to(v),cap(c),flow(f){}
};

struct Dinic
{
    int n,m,s,t;//結點數,邊數(包括反向弧),源點編號,匯點編號
    vector<Edge>edges;//邊表,dges[e]和dges[e^1]互為反向弧
    vector<int>G[maxn];//鄰接表,G[i][j]表示結點i的第j條邊在e陣列中的編號
    bool vis[maxn]; //BFS的使用
    int d[maxn]; //從起點到i的距離
    int cur[maxn]; //當前弧下標
    
    void addedge(int from,int to,int cap)
    {
        edges.push_back(Edge(from,to,cap,0));
        edges.push_back(Edge(to,from,0,0));
        ULL  m=edges.size();
        G[from].push_back(m-2);
        G[to].push_back(m-1);
    }
    
    bool bfs()
    {
        memset(vis,0,sizeof(vis));
        queue<int>Q;
        Q.push(s);
        d[s]=0;
        vis[s]=1;
        while(!Q.empty())
        {
            int x=Q.front();Q.pop();
            for(int i=0;i<G[x].size();i++)
            {
                Edge&e=edges[G[x][i]];
                if(!vis[e.to]&&e.cap>e.flow)//只考慮殘量網路中的弧
                {
                    vis[e.to]=1;
                    d[e.to]=d[x]+1;
                    Q.push(e.to);
                }
            }
            
        }
        return vis[t];
    }
    
    int dfs(int x,int a)//x表示當前結點,a表示目前為止的最小殘量
    {
        if(x==t||a==0)return a;//a等於0時及時退出,此時相當於斷路了
        int flow=0,f;
        for(int&i=cur[x];i<G[x].size();i++)//從上次考慮的弧開始,注意要使用引用,同時修改cur[x]
        {
            Edge&e=edges[G[x][i]];//e是一條邊
            if(d[x]+1==d[e.to]&&(f=dfs(e.to,min(a,e.cap-e.flow)))>0)
            {
                e.flow+=f;
                edges[G[x][i]^1].flow-=f;
                flow+=f;
                a-=f;
                if(!a)break;//a等於0及時退出,當a!=0,說明當前節點還存在另一個曾廣路分支。
                
            }
        }
        return flow;
    }
    
    int Maxflow(int s,int t)//主過程
    {
        this->s=s; this->t=t;
        int flow=0;
        while(bfs())//不停地用bfs構造分層網路,然後用dfs沿著阻塞流增廣
        {
            memset(cur,0,sizeof(cur));
            flow+=dfs(s,INF);
        }
        return flow;
    }
};