首页 > 代码库 > 最大流dinci模板
最大流dinci模板
我们知道,增广路EK算法的时间负责度是O(n*m^2),找最短增广路的时间复杂度是O(m*n),所以时间复杂度主要是在找增广路上。
这里介绍另一种Dinci算法,用BFS构造层次图,然后用DFS增广。
模板
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <algorithm> #include <vector> #include <queue> using namespace std; #define Del(a,b) memset(a,b,sizeof(a)) const int N = 1000; const int INF = 0x3f3f3f3f; struct Edge { int from,to,cap,flow; }; struct Dinic { int n,m,s,t; vector<Edge> edges; vector<int> G[N]; bool vis[N]; int d[N],cur[N]; void init(int n) { this->n=n; for(int i=0;i<=n;i++)G[i].clear(); edges.clear(); } void AddEdge(int from,int to,int cap){ //建边 edges.push_back((Edge){from,to,cap,0}); edges.push_back((Edge){to,from,0,0}); m=edges.size(); G[from].push_back(m-2); G[to].push_back(m-1); } bool BFS() { Del(vis,0); 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) { if(x==t || a==0) return a; int flow=0,f; for(int& i=cur[x];i<G[x].size();i++) { Edge & e = edges[G[x][i]]; 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==0) break; } } return flow; } int max_flow(int s,int t) { this->s=s;this->t=t; int flow=0; while(BFS()) { Del(cur,0); flow+=DFS(s,INF); } return flow; } }; Dinic solve; int main() { int n,m; while(~scanf("%d%d",&m,&n)) { solve.init(n); for(int i=0; i<m; i++) { int x,y,z; scanf("%d%d%d",&x,&y,&z); solve.AddEdge(x,y,z); } printf("%d\n",solve.max_flow(1,n)); //出发点和结尾点 } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。