hdu 1532 Dinic模板（小白书）

hdu1532

输入n，m。

n条边，m个点，之后给出a到b的容量，求1到m的最大流。

注意：Dinic只能调用一次，因为原理是改变cap的值，如果调用多次一样的，那么第一次会对，其余的都会是0，因为，cap的值经过一次调用已经改变了，如果想调用多次，就只能再开一个数组，存下原来的cap值。

#include <bits/stdc++.h>
using namespace std;

const int INF=0x3f3f3f3f;
typedef long long LL;
#define PI(A) printf("%d\n",A)
#define SI(N) scanf("%d",&(N))
#define SII(N,M) scanf("%d%d",&(N),&(M))
#define cle(a,val) memset(a,(val),sizeof(a))
#define rep(i,b) for(int i=0;i<(b);i++)
#define Rep(i,a,b) for(int i=(a);i<=(b);i++)
#define reRep(i,a,b) for(int i=(a);i>=(b);i--)
const double EPS= 1e- ;

/*  /////////////////////////     C o d i n g  S p a c e     /////////////////////////  */

/** \brief Dinic 模板
 *
 * \param MAX_V是最大顶点
 * \param s是开始，t是结束
 * \return 返回int型答案
 * ps：多组输入要初始化
 */     

const int MAX_V=  +  ;

struct edge{int to,cap,rev;};

vector<edge> G[MAX_V];
int level[MAX_V];
int iter[MAX_V];

void add_edge(int from,int to,int cap)
{
    G[from].push_back((edge){to,cap,G[to].size()});
    G[to].push_back((edge){from,,G[from].size()-});
}

void bfs(int s)
{
    memset(level,-,sizeof(level));
    queue<int> que;
    level[s]=;
    que.push(s);
    while(!que.empty())
    {
        int v=que.front();que.pop();
        for (int i=;i<G[v].size();i++)
        {
            edge& e=G[v][i];
            if (e.cap>&&level[e.to]<)
            {
                level[e.to]=level[v]+;
                que.push(e.to);
            }
        }
    }
}

int dfs(int v,int t,int f)
{
    if (v==t) return f;
    for (int &i=iter[v];i<G[v].size();i++)///这的i一定是引用，因为要改变iter[v]的值
    {
        edge& e=G[v][i];
        if (e.cap>&&level[v]<level[e.to])
        {
            int d=dfs(e.to,t,min(f,e.cap));
            if (d>)
            {
                e.cap-=d;
                G[e.to][e.rev].cap+=d;
                return d;
            }
        }
    }
    return ;
}

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

}

///end///////////////////////////////////////////

int main()
{
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Zmy\\Desktop\\in.txt","r",stdin);
//    freopen("C:\\Users\\Zmy\\Desktop\\out.txt","w",stdout);
#endif
iostream::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);

int n,m;
while(cin>>n>>m)
{

    for (int i=;i<MAX_V;i++) G[i].clear();///多组输入一定要初始化

    int ans=;
    for (int i=;i<n;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        add_edge(u,v,w);
    }
    ans=max_flow(,m);
    printf("%d\n",ans);
}
 return ;
}