Advertisement
Guest User

Untitled

a guest
Apr 8th, 2020
274
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.25 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <algorithm>
  4. #include <cstdio>
  5. #include <string>
  6. #include <queue>
  7. #include <stack>
  8. #include <set>
  9. #include <unordered_map>
  10. #include <unordered_set>
  11. #include <sstream>
  12. #include <math.h>
  13. #include <map>
  14.  
  15. using namespace std;
  16.  
  17. int n, m, s, t;
  18. unsigned long long flow = 0;
  19. vector<unordered_map<int, pair<int, int>>> g;
  20. vector<bool> mark;
  21.  
  22. int dfs(int v, int d) {
  23.     if (mark[v])
  24.         return 0;
  25.     mark[v] = true;
  26.     if (v == t) {
  27.         return d;
  28.     }
  29.     for (auto iter = g[v].begin(); iter != g[v].end(); iter++) {
  30.         if (iter->second.second > iter->second.first) {
  31.             int u = iter->first;
  32.             int res = dfs(iter->first, min(d, iter->second.second - iter->second.first));
  33.             if (res > 0) {
  34.                 g[v][u].first += res;
  35.                 g[u][v].first -= res;
  36.                 return res;
  37.             }
  38.         }
  39.     }
  40.     return 0;
  41. }
  42.  
  43.  
  44. int bfs() {
  45.     fill(mark.begin(), mark.end(), false);
  46.     queue<pair<int, int>> q = {};
  47.     q.push({-1, s});
  48.     vector<int> prv(n , -1);
  49.  
  50.     while (!q.empty()) {
  51.         int v = q.front().second;
  52.         prv[v] = q.front().first;
  53.         q.pop();
  54.         mark[v] = true;
  55.         if (v == t) {
  56.             break;
  57.         }
  58.         for (auto &iter : g[v]) {
  59.             if (!mark[iter.first] && iter.second.second > iter.second.first) {
  60.                 q.push({v, iter.first});
  61.             }
  62.         }
  63.     }
  64.  
  65.     int cur = t;
  66.     int d = INT_MAX;
  67.     if (prv[t] == -1) {
  68.         return 0;
  69.     }
  70.  
  71.     while (cur != s) {
  72.         d = min(d, g[prv[cur]][cur].second - g[prv[cur]][cur].first);
  73.         cur = prv[cur];
  74.     }
  75.  
  76.     cur = t;
  77.     while (cur != s) {
  78.         g[prv[cur]][cur].first += d;
  79.         g[cur][prv[cur]].first -= d;
  80.         cur = prv[cur];
  81.     }
  82.  
  83.     return d;
  84. }
  85.  
  86.  
  87. unordered_set<int> A = {};
  88. set<pair<int, int>> ans = {};
  89.  
  90. void find_cut() {
  91.     queue<int> q = {};
  92.     q.push(s);
  93.     while (!q.empty()) {
  94.         int v = q.front();
  95.         q.pop();
  96.         for (auto &iter : g[v]) {
  97.             if (A.find(iter.first) == A.end()) {
  98.                 if (iter.second.second > iter.second.first) {
  99.                     A.insert(iter.first);
  100.                     q.push(iter.first);
  101.                 } else {
  102.                     ans.insert({v, iter.first});
  103.                 }
  104.             }
  105.         }
  106.     }
  107. }
  108.  
  109. map<pair<int, int>, int> my_map;
  110.  
  111. int main() {
  112.     cin >> n >> m;
  113.     s = 0;
  114.     t = n - 1;
  115.     g.resize(n);
  116.     mark.resize(n);
  117.  
  118.  
  119.     for (int i = 0; i < m; ++i) {
  120.         int u, v, c;
  121.         cin >> u >> v >> c;
  122.         u--;
  123.         v--;
  124.         g[u][v].second = c;
  125.         g[v][u].second = c;
  126.         my_map[{u, v}] = i;
  127.         my_map[{v, u}] = i;
  128.     }
  129.  
  130.     while (true) {
  131. //        fill(mark.begin(), mark.end(), false);
  132. //        int delta = dfs(s, INT_MAX);
  133.         int delta = bfs();
  134.         if (delta > 0) {
  135.             flow += delta;
  136.         } else {
  137.             break;
  138.         }
  139.     }
  140.  
  141.     A.insert(s);
  142.     find_cut();
  143.     cout << ans.size() << ' ' << flow << endl;
  144.     for (auto &e : ans) {
  145.         if (A.find(e.first) == A.end() || A.find(e.second) == A.end()) {
  146.             cout << my_map[e] + 1 << ' ';
  147.         }
  148.     }
  149. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement