fork(1) download
copy 
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. using ll = long long;
  4. using vl = vector<ll>;
  5. #define FOR(i, a, b) for(ll i = (a); i < (b); i++)
  6. #define FORD(i, a, b) for(ll i = (b)-1; i >= (a); i--)
  7.  
  8. typedef vector<int> VI;
  9. typedef vector<VI> VVI;
  10. const ll INF = 1000000000000000000LL;
  11.  
  12. #define VEI(w,e) ((E[e].u == w) ? E[e].v : E[e].u)
  13. #define CAP(w,e) ((E[e].u == w) ? E[e].cap[0] - E[e].flow : E[e].cap[1] + E[e].flow)
  14. #define ADD(w,e,f) E[e].flow += ((E[e].u == w) ? (f) : (-(f)))
  15.  
  16. struct Edge { int u, v; ll cap[2], flow; };
  17.  
  18. VI d, act;
  19.  
  20. bool bfs(int s, int t, VVI& adj, vector<Edge>& E) {
  21.   queue<int> Q;
  22.   d = VI(adj.size(), -1);
  23.   d[t] = 0;
  24.   Q.push(t);
  25.   while (not Q.empty()) {
  26.     int u = Q.front(); Q.pop();
  27.     for (int i = 0; i < int(adj[u].size()); ++i) {
  28.       int e = adj[u][i], v = VEI(u, e);
  29.       if (CAP(v, e) > 0 and d[v] == -1) {
  30.         d[v] = d[u] + 1;
  31.         Q.push(v);
  32.       }
  33.     }
  34.   }
  35.   return d[s] >= 0;
  36. }
  37.  
  38. ll dfs(int u,int t,ll bot,VVI& adj,vector<Edge>& E) {
  39.   if (u == t) return bot;
  40.   for (; act[u] < int(adj[u].size()); ++act[u]) {
  41.     int e = adj[u][act[u]];
  42.     if (CAP(u, e) > 0 and d[u] == d[VEI(u, e)] + 1) {
  43.       ll inc=dfs(VEI(u,e),t,min(bot,CAP(u,e)),adj,E);
  44.       if (inc) {
  45.         ADD(u, e, inc);
  46.         return inc;
  47.       }
  48.     }
  49.   }
  50.   return 0;
  51. }
  52.  
  53. ll maxflow(int s, int t, VVI& adj, vector<Edge>& E) {
  54.   for (int i=0; i<int(E.size()); ++i) E[i].flow = 0;
  55.   ll flow = 0, bot;
  56.   while (bfs(s, t, adj, E)) {
  57.     act = VI(adj.size(), 0);
  58.     while ((bot = dfs(s,t,INF, adj, E))) flow += bot;
  59.   }
  60.   return flow;
  61. }
  62.  
  63. void addEdge(int u, int v, VVI& adj, vector<Edge>& E, ll cap){
  64. 	Edge e;
  65. 	e.u = u;
  66. 	e.v = v;
  67. 	e.cap[0] = cap;
  68. 	e.cap[1] = cap;
  69. 	e.flow = 0;
  70. 	adj[u].push_back(E.size());
  71. 	adj[v].push_back(E.size());
  72. 	E.push_back(e);
  73. }
  74. int main() {
  75. 	int n, m;
  76. 	cin >> n >> m;
  77. 	vector<vector<int>>adj(n);
  78. 	vector<Edge> E;
  79. 	for(int i = 0; i < m; ++i) {
  80. 		int x,y,w;
  81. 		cin >> x >> y >> w;
  82. 		--x;--y;
  83. 		addEdge(x, y, adj, E, w);
  84. 	}
  85. 	cout << maxflow(0, n-1, adj ,E) << endl;
  86. }
Success #stdin #stdout 0s 15248KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
0
https://ideone.com/jkLTyV
language:
C++ (gcc 8.3)
created:
7 years ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!