【模板】Dinic算法

#include<vector>
#include<cstring>
#include<algorithm>
#include<queue>
#define INF 0x7fffffff
#define MAX_V 210

using namespace std;

//用于表示边的结构体
struct edge
{
    int to;    //终点
    int cap;   //容量
    int rev;   //反向边
};

int m,n;
int iter[MAX_V];         //顶点到源咪的距离标号
int level[MAX_V];        //当前弧，在其之前的边已经没有用了
vector<edge> G[MAX_V];   //图的邻接表表示，多组测试数据时注意初始化（清空）

//向图中增加一条从from到to的容量为cap的边
void add_edge(int _from,int _to,int _cap)
{
    edge temp;
    temp.to=_to;
    temp.cap=_cap;
    temp.rev=G[_to].size();
    G[_from].push_back(temp);
    temp.to=_from;
    temp.cap=0;
    temp.rev=G[_from].size()-1;
    G[_to].push_back(temp);
}

//通过BFS计算从源点出发的距离标号
void bfs(int s)
{
    memset(level,-1,sizeof(level));
    queue<int> que;
    level[s]=0;
    que.push(s);
    while(!que.empty())
    {
        int v=que.front();
        que.pop();
        for(int i=0;i<G[v].size();i++)
        {
            edge &e=G[v][i];
            if(e.cap>0&&level[e.to]<0)
            {
                level[e.to]=level[v]+1;
                que.push(e.to);
            }
        }
    }
}

//通过DFS寻找增广路
int dfs(int v,int t,int f)
{
    if(v==t)
        return f;

    for(int &i=iter[v];i<G[v].size();i++)
    {
        edge &e=G[v][i];
        if(e.cap>0&&level[v]<level[e.to])
        {
            int d=dfs(e.to,t,min(f,e.cap));
            if(d>0)
            {
                e.cap-=d;
                G[e.to][e.rev].cap+=d;
                return d;
            }
        }
    }

    return 0;
}

//求解从s到t的最大流
int max_flow(int s,int t)
{
    int flow=0;

    while(true)
    {
        bfs(s);
        if(level[t]<0)
            return flow;
        memset(iter,0,sizeof(iter));
        int f;
        while((f=dfs(s,t,INF))>0)
            flow+=f;
    }
}