Advertisement
Mohammad_Dipu_Sultan

Heavy Cycle Edges UVA - 11747 || MST

Sep 30th, 2021
753
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 6.62 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3.  
  4. typedef long long           ll;
  5. typedef unsigned long long  ull;
  6. typedef double              ld;//%Lf
  7. #define ff                  first
  8. #define ss                  second
  9. #define pb                  push_back
  10. #define mp                  make_pair
  11. #define ppb                 pop_back
  12. #define in                  insert
  13. #define sz(a)               int((a).size())
  14. #define vi                  vector<int>
  15. #define vl                  vector<ll>
  16. #define vvi                 vector<vector<int> >
  17. #define pii                 pair<int,int>
  18. #define piii                pair<int,pair<int,int> >
  19. #define pll                 pair<ll,ll>
  20. #define vii                 vector<pii>
  21. #define vll                 vector<pll>
  22. #define viii                vector<piii>
  23. #define YES                 cout << "YES\n"
  24. #define NO                  cout << "NO\n"
  25.  
  26. #define scl(n)              scanf("%lld",&n)
  27. #define scll(x,y)           scanf("%lld %lld",&x,&y)
  28. #define gtl(x)              getline(cin, (x))
  29.  
  30. #define f0(b)               for(int i=0;i<(b);i++)
  31. #define f00(b)              for(int j=0;j<(b);j++)
  32. #define f1(b)               for(int i=1;i<=(b);i++)
  33. #define f11(b)              for(int j=1;j<=(b);j++)
  34. #define f2(a,b)             for(int i=(a);i<=(b);i++)
  35. #define f22(a,b)            for(int j=(a);j<=(b);j++)
  36. #define RFOR(i,x,y)         for(int i=x;i>=y;i--)
  37. #define all(v)              v.begin(),v.end()
  38. #define rall(v)             v.rbegin(),v.rend()
  39. #define unq(v)              sort(all(v)),(v).resize(unique(all(v))-v.begin())
  40. #define present(c,x)        ((c).find(x) != (c).end())
  41. #define cpresent(c,x)       (find(all(c),x) != (c).end())
  42. #define min_ele(v)          (*min_element(all(v)))
  43. #define max_ele(v)          (*max_element(all(v)))
  44. #define cnt_ele(v, x)       (count(all(v), x))
  45. #define sum_ele(v)          (accumulate(all(v),0))
  46. #define reversed(s)         reverse(s.begin(), s.end())
  47.  
  48. #define testcase            ll t; cin>>t; while (t--)
  49. #define vout(v)             for(int ind=0;ind<v.size();ind++){ cout<<v[ind]; if(ind<v.size()-1) cout<<' '; else cout<<endl;}
  50. #define arrout(arr,i,x,y)   for(int i=x;i<=y;i++){ cout<<arr[i]; if(i<y) cout<<' '; else cout<<endl;}
  51. #define PI                  acos(-1)
  52. #define CLR(x, y)           memset(x, y, sizeof(x))
  53. #define Precision(a)        cout << fixed << setprecision(a)
  54. #define BITCH_FAST()        ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0) //Don't use scanf/printf
  55. #define MAXP                1000006
  56. #define MOD                 1e9+7//10000007 // For big mod
  57. #define EPS                 1e-7
  58.  
  59. template <typename T> T Sqr(T x) { T n = x * x ; return n ;}
  60. template <typename T> T Pow(T B,T P){ if(P==0) return 1; if(P&1) return B*Pow(B,P-1);  else return Sqr(Pow(B,P/2));}
  61. template <typename T> T Abs(T a) {if(a<0)return -a;else return a;}
  62. template <typename T> T Gcd(T a,T b){if(a<0)return Gcd(-a,b);if(b<0)return Gcd(a,-b);return (b==0)?a:Gcd(b,a%b);} // better than __gcd
  63. template <typename T> T Lcm(T a,T b) {if(a<0)return Lcm(-a,b);if(b<0)return Lcm(a,-b);return a*(b/Gcd(a,b));}
  64. template <typename T> T power(T n, T p){ T res = 1;while(p > 0){if(p & 1)res *= n;n *= n;p >>= 1;}return res;}
  65. template <typename T> T BigMod (T b,T p,T m){if (p == 0) return 1;if (p%2 == 0){T s = BigMod(b,p/2,m);return ((s%m)*(s%m))%m;}return ((b%m)*(BigMod(b,p-1,m)%m))%m;}
  66. template <typename T> inline string ToBinary(T n) {string r ;while(n != 0) {r = ( n % 2 == 0 ? "0" : "1" ) + r ; n >>= 1;} return r ;}
  67. long long BinaryToDecimal(string s) {int len = s.size();long long n = 0, p = 1;for (int i = len - 1; i >= 0; i-- , p *= 2) n += p * (s[i] - '0');return n;}
  68. int Strtoint(string str){stringstream ss(str);int x = 0;ss >> x ;return x ;}
  69. string Intostr(int x){stringstream ss; ss << x; string str = ss.str(); return str;}
  70.  
  71. #define TRACE
  72. #ifdef TRACE
  73. #define trace(...) __f(#__VA_ARGS__, __VA_ARGS__)
  74.     template <typename Arg1>
  75.     void __f(const char* name, Arg1&& arg1){
  76.         cout << name << " : " << arg1 << endl;
  77.         //use cerr if u want to display at the bottom
  78.     }
  79.     template <typename Arg1, typename... Args>
  80.     void __f(const char* names, Arg1&& arg1, Args&&... args){
  81.         const char* comma = strchr(names + 1, ','); cout.write(names, comma - names) << " : " << arg1<<" | ";__f(comma+1, args...);
  82.     }
  83. #else
  84. #define trace(...)
  85. #endif
  86.  
  87. /*----------------------Graph Moves----------------*/
  88. int ROW[]={+1,-1,+0,+0};
  89. int COL[]={+0,+0,+1,-1};
  90.  
  91. int X[]={+0,+0,+1,-1,-1,+1,-1,+1};   // Kings Move
  92. int Y[]={-1,+1,+0,+0,+1,+1,-1,-1};  // Kings Move
  93.  
  94. int KX[]={-2, -2, -1, -1,  1,  1,  2,  2};  // Knights Move
  95. int KY[]={-1,  1, -2,  2, -2,  2, -1,  1}; // Knights Move
  96. map<int, int>MP;
  97. vi parent, raank;
  98. viii edges;
  99. void makeset(int n)
  100. {
  101.     for(int i=0; i<n; i++)
  102.     {
  103.         parent[i]=i;
  104.         raank[i]=0;
  105.     }
  106. }
  107. int find_parent(int v)
  108. {
  109.     if(parent[v]==v)
  110.     {
  111.         return v;
  112.     }
  113.     return parent[v]=find_parent(parent[v]);
  114. }
  115. void union_seet(int u, int v)//Join
  116. {
  117.     u=find_parent(u);
  118.     v=find_parent(v);
  119.     if(u!=v)
  120.     {
  121.         if(rand()%2)
  122.         {
  123.             swap(u, v);
  124.         }
  125.         parent[v]=u;
  126.     }
  127. }
  128.  
  129. void Kruskals(viii &edges, int n)
  130. {
  131.     sort(all(edges));
  132.     vector<int>v;
  133.     v.clear();
  134.     for(auto e:edges)
  135.     {
  136.         int u=e.ss.ff;
  137.         int v=e.ss.ss;
  138.         int set_u=find_parent(u);
  139.         int set_v=find_parent(v);
  140.         if(set_u!=set_v)
  141.         {
  142.             union_seet(u,v);
  143.             //wt+=e.ff;
  144.             MP[e.ff]--;
  145.             //trace(e.ff);
  146.  
  147.         }
  148.     }
  149.     for(auto ex:MP)
  150.     {
  151.         if(ex.ss)
  152.         {
  153.             while(ex.ss--)
  154.             {
  155.                 v.pb(ex.ff);
  156.             }
  157.         }
  158.     }
  159.     if(sz(v)==0) cout << "forest\n";
  160.     else
  161.     {
  162.         f0(sz(v))
  163.         {
  164.             if(i==sz(v)-1) cout << v[i] << endl;
  165.             else cout << v[i] << " ";
  166.         }
  167.     }
  168. }
  169. int main()
  170. {
  171.     //BITCH_FAST();
  172.     #ifdef FLAME
  173.         clock_t tStart = clock();
  174.         freopen("input.txt", "r", stdin);
  175.         freopen("output.txt", "w", stdout);
  176.     #endif
  177.     ll n, m;
  178.     while(cin >> n >> m, !(n==0 && m==0))
  179.     {
  180.         edges.clear();
  181.         MP.clear();
  182.         parent.resize(n);
  183.         raank.resize(n);
  184.         makeset(n);
  185.         f0(m)
  186.         {
  187.             int a, b, x;
  188.             cin >> a >> b >> x;
  189.             MP[x]++;
  190.             edges.push_back({x,{a, b}});
  191.            
  192.  
  193.         }
  194.         Kruskals(edges, m);
  195.     }
  196.     #ifdef FLAME
  197.         fprintf(stderr, "\n>> Runtime: %.10fs\n", (double) (clock() - tStart) / CLOCKS_PER_SEC);
  198.     #endif
  199.     return 0;
  200. }
  201.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement