El_GEMMY

calling circles uva (using tarjan scc)

Jul 27th, 2022
77
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 5.00 KB | None | 0 0
  1. // Those who cannot remember the past are
  2. // condemned to repeat it (use DP -_-)
  3. // - George Santayana
  4.  
  5. #include <bits/stdc++.h>
  6. #include <ext/pb_ds/assoc_container.hpp>
  7. #include <ext/pb_ds/tree_policy.hpp>
  8.  
  9. using namespace std;
  10. using namespace __gnu_pbds;
  11.  
  12. #define all(v) v.begin(), v.end()
  13. #define rall(v) v.rbegin(), v.rend()
  14. #define ll long long
  15. #define ull unsigned long long
  16. #define MOD 1000000007
  17. #define PI acos(-1)
  18. //#define ceil(a, b) (((a) / (b)) + ((a) % (b) ? 1 : 0))
  19. #define imin INT_MIN
  20. #define imax INT_MAX
  21. #define llmax LLONG_MAX
  22. #define llmin LLONG_MIN
  23. #define inf 2000000000
  24. #define nl '\n'
  25. #define ppcnt __builtin_popcount
  26. #define ppcntll __builtin_popcountll
  27. #define clz __builtin_clz
  28. #define clzll __builtin_clzll
  29. #define ctz __builtin_ctz
  30. #define ctzll __builtin_ctzll
  31. #define modulo(a, b, mod) ((((a) % (mod)) + ((b) % (mod))) % (mod))
  32. #define cnte(v, x) count(all(v), (x))
  33. #define mine(v) min_element(all(v))
  34. #define maxe(v) max_element(all(v))
  35. #define updmin(a, b) a = min(a, b)
  36. #define updmax(a, b) a = max(a, b)
  37. #define findmod(x, m) x = ((x) % (m) + (m)) % m
  38. #define getmod(x, m) ((x) % (m) + (m)) % (m)
  39. #define debug(x) cout << "x: " << (x) << nl;
  40. #define debug2(x, y) cout << "x: " << (x) << " y: " << y << nl;
  41. #define ordered_set tree<pair<int, int>, null_type, less<>, rb_tree_tag, tree_order_statistics_node_update>
  42. #define ordered_map tree<int, int, less<>, rb_tree_tag, tree_order_statistics_node_update>
  43.  
  44. //vector<int> dx = {0, 0, 1, -1, 1, 1, -1, -1}, dy = {1, -1, 0, 0, 1, -1, 1, -1};
  45. vector<int> dx = {0, 0, 1, -1}, dy = {1, -1, 0, 0};
  46.  
  47. template<typename T = int>
  48. istream &operator>>(istream &in, vector<pair<int, int>> &v) {
  49.     for (auto &[x, y]: v) in >> x >> y;
  50.     return in;
  51. }
  52.  
  53. template<typename T = int>
  54. istream &operator>>(istream &in, vector<T> &v) {
  55.     for (T &i: v) in >> i;
  56.     return in;
  57. }
  58.  
  59. template<typename T = int>
  60. ostream &operator<<(ostream &out, const vector<T> &v) {
  61.     for (const T &x: v)
  62.         out << x << ' ';
  63.     return out;
  64. }
  65.  
  66. template<typename T = pair<int, int>>
  67. ostream &operator<<(ostream &out, const vector<pair<int, int>> &v) {
  68.     for (auto &[x, y]: v) {
  69.         out << x << ' ' << y << nl;
  70.     }
  71.     return out;
  72. }
  73.  
  74. void Start_Crushing() {
  75.     ios_base::sync_with_stdio(false);
  76.     cin.tie(nullptr);
  77.     cout.tie(nullptr);
  78. #ifndef ONLINE_JUDGE
  79.     freopen("input.txt", "r", stdin);
  80.     freopen("output.txt", "w", stdout);
  81. #endif
  82. }
  83.  
  84. map<string, vector<string>> adj;
  85. map<string, int> dfs_num, dfs_low;
  86. int counter;
  87. stack<string> order;
  88. map<string, bool> vis;
  89. vector<vector<string>> sccs;
  90.  
  91. void tarjan_scc(string& src){
  92.     dfs_num[src] = dfs_low[src] = counter++;
  93.     order.push(src);
  94.     vis[src] = true;
  95.  
  96.     for(auto& nxt : adj[src]){
  97.         if(not dfs_num[nxt]){
  98.             tarjan_scc(nxt);
  99.         }
  100.         if(vis[nxt])
  101.             dfs_low[src] = min(dfs_low[src], dfs_low[nxt]);
  102.     }
  103.     if(dfs_low[src] == dfs_num[src]){
  104.  
  105.         vector<string> scc;
  106.  
  107.         while(not order.empty()){
  108.             scc.push_back(order.top());
  109.             order.pop();
  110.             vis[scc.back()] = false;
  111.             if(scc.back() == src)
  112.                 break;
  113.         }
  114.  
  115.         sccs.emplace_back(scc);
  116.     }
  117. }
  118.  
  119. void solve(int tc) {
  120.     // we wanna find all the SCCs and count only the SCC which is greater than 2
  121.  
  122.     int n, m; cin >> n >> m;
  123.     while(n and m){
  124.         adj.clear();
  125.         dfs_num.clear(), dfs_low.clear(), vis.clear(), sccs.clear();
  126.         counter = 1;
  127.  
  128.         vector<string> people;
  129.  
  130.         for(int i = 0; i < m; i++){
  131.             string u, v; cin >> u >> v;
  132.             adj[u].emplace_back(v);
  133.             if(not vis[u]) {
  134.                 people.emplace_back(u);
  135.                 vis[u] = true;
  136.             }
  137.             if(not vis[v]) {
  138.                 people.emplace_back(v);
  139.                 vis[v] = true;
  140.             }
  141.         }
  142.         vis.clear();
  143.  
  144.         for(auto& person : people){
  145.             if(not dfs_num[person])
  146.                 tarjan_scc(person);
  147.         }
  148.  
  149.         cout << "Calling circles for data set " << tc++ << ":\n";
  150.         for(auto& scc : sccs){
  151.             for(int i = 0; i < scc.size(); i++){
  152.                 if(i)
  153.                     cout << ", ";
  154.                 cout << scc[i];
  155.             }
  156.             cout << nl;
  157.         }
  158.         cout << nl;
  159.         cin >> n >> m;
  160.     }
  161. }
  162.  
  163. void set_file(string &file_name) {
  164.     freopen((file_name + ".in").c_str(), "r", stdin);
  165.     freopen((file_name + ".out").c_str(), "w", stdout);
  166. }
  167.  
  168. int main() {
  169.     Start_Crushing();
  170.  
  171. //    string file_name = "";
  172. //    set_file(file_name);
  173.  
  174.     int t = 1;
  175. //    /*Multiple test cases?*/ cin >> t;
  176.     while (t--) {
  177.         solve(1);
  178. //        if (!t)
  179. //            break;
  180. //        cout << nl;
  181.     }
  182.  
  183. //    for(int tc = 1; tc <= t; tc++){
  184. ////        cout << "Calling circles for data set " << tc << ":\n";
  185. //        solve(tc);
  186. ////        if(tc != t)
  187. ////            cout << nl;
  188. //        printf("\n");
  189. //    }
  190.  
  191.     return 0;
  192. }
Advertisement
Add Comment
Please, Sign In to add comment