最大流演算法模板 挑戰程式設計競賽
阿新 • • 發佈:2019-01-10
const int MAX_V=1e6; struct edge{ int to,cap,rev; //終點,容量,反向邊 }; vector<edge> G[MAX_V]; //圖的鄰接表表示 int level[MAX_V]; //頂點到原點的距離標號 int iter[MAX_V]; //當前弧,在其之前的邊已經沒有用了 //增加一條從from到to的容量為cap的邊 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,0,G[from].size()-1}); } //通過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; for(;;){ bfs(s); if(level[t]<0) return flow; memset(iter,0,sizeof(iter)); int f; while((f=dfs(s,t,INF))>0){ flow+=f; } } }