Advertisement
Korotkodul

CF

Sep 10th, 2022 (edited)
95
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.04 KB | None | 0 0
  1. #include <iostream>
  2. #include <cmath>
  3. #include <vector>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <string>
  7. #include <stack>
  8. #include <set>
  9. #include <map>
  10. #define pii pair <int,int>
  11. #define vec vector
  12. using namespace std;
  13. using ll = long long;
  14. using ld = long double;
  15. using db = double;
  16. void cv(vector <int> &v){
  17.     for (auto x: v) cout<<x<<' ';
  18.     cout<<"\n";
  19. }
  20.  
  21. void cvl(vector <ll> &v){
  22.     for (auto x: v) cout<<x<<' ';
  23.     cout<<"\n";
  24. }
  25.  
  26.  
  27. void cvv(vector <vector <int> > &v){
  28.     for (auto x: v) cv(x);
  29.     cout<<"\n";
  30. }
  31.  
  32. void cvb(vector <bool> v){
  33.     for (bool x: v) cout<<x<<' ';
  34.     cout<<"\n";
  35. }
  36.  
  37. void cvs(vector <string>  v){
  38.     for (auto a: v){
  39.         cout<<a<<"\n";
  40.     }
  41. }
  42.  
  43. void cvp(vector <pii> a){
  44.     for (auto p: a){
  45.         cout<<p.first<<' '<<p.second<<"\n";
  46.     }
  47.     cout<<"\n";
  48. }
  49. int n,m,A,B;
  50. struct ed{
  51.     int a,b,l;
  52. };
  53. vector <ed> E;
  54.  
  55. vector <int> pr,sz;
  56.  
  57. bool sh = 1;
  58.  
  59. int getp(int v){
  60.     if (pr[v] == v){
  61.         return v;
  62.     }
  63.     int p = getp(v);
  64.     pr[v] = p;
  65.     return p;
  66. }
  67.  
  68. void un(int a, int b){
  69.     a = getp(a);
  70.     b = getp(b);
  71.     if (a == b)return;
  72.     //sz[b] > sz[a]
  73.     if (sz[a] > sz[b]) swap(a,b);
  74.     sz[b] += sz[a];
  75.     pr[a] = b;
  76. }
  77.  
  78. int main()
  79. {
  80.     ios::sync_with_stdio(0);
  81.     cin.tie(0);
  82.     cout.tie(0);
  83.     cin>>n>>m;
  84.     map <pii, int> mtr;
  85.     for (int i = 0; i < m; ++i){
  86.             int a,b,l;
  87.             cin>>a>>b>>l;
  88.             if (a > b) swap(a,b);
  89.             if (mtr.find({a,b}) == mtr.end()){
  90.                 mtr[{a,b}] = l;
  91.             }
  92.             else{
  93.                 mtr[{a,b}] = max(mtr[{a,b}], l);
  94.             }
  95.     }
  96.     cin>>A>>B;
  97.     for (auto p: mtr){
  98.         //cout<<p.first.first<<' '<<p.first.second<<' '<<p.second<<"\n";
  99.         E.push_back({p.first.first, p.first.second, p.second});
  100.     }
  101.     pr.resize(n);
  102.     sz.assign(n,1);
  103.     for (int i = 0; i < n; ++i){
  104.         pr[i] = i;
  105.     }
  106.     for (ed e: E){
  107.         un(e.a, e.b);
  108.     }
  109.     if (sh){
  110.         cout<<"sz\n";
  111.         cv(sz);
  112.         cout<<"pr\n";
  113.         cv(pr);
  114.     }
  115.  
  116. }
  117.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement