Advertisement
7oSkaaa

Dream Team

Sep 23rd, 2021
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.87 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define cin(vec) for(auto& i : vec) cin >> i
  5. #define cin_2d(vec, n, m) for(int i = 0; i < n; i++) for(int j = 0; j < m && cin >> vec[i][j]; j++);
  6. #define cout(vec) for(auto& i : vec) cout << i << " "; cout << "\n";
  7. #define cout_2d(vec, n, m) for(int i = 0; i < n; i++, cout << "\n") for(int j = 0; j < m && cout << vec[i][j] << " "; j++);
  8. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << "  " << s << "\n";
  9. #define Time cerr << "Time Taken: " << (float)clock() / CLOCKS_PER_SEC << " Secs" << "\n";
  10. #define fixed(n) fixed << setprecision(n)
  11. #define ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  12. #define fill(vec, value) memset(vec, value, sizeof(vec));
  13. #define Num_of_Digits(n) ((int)log10(n)+1)
  14. #define mod_combine(a, b, m) (((a % m) * (b % m)) % m)
  15. #define all(vec) vec.begin(),vec.end()
  16. #define rall(vec) vec.rbegin(),vec.rend()
  17. #define sz(x) int(x.size())
  18. #define TC int t; cin >> t;   while(t--)
  19. #define fi first
  20. #define se second
  21. #define Pair pair < int, int >
  22. #define ll long long
  23. #define ull unsigned long long
  24. #define Mod  1'000'000'007
  25. #define OO 2'000'000'000
  26. #define EPS 1e-9
  27. #define PI acos(-1)
  28.  
  29. void AhMeD_HoSSaM(){
  30.   ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  31.   #ifndef ONLINE_JUDGE
  32.     //freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  33.   #endif
  34.   freopen("dream.in", "r", stdin);
  35. }
  36. struct DSU {
  37.     vector < int > parent, Gsize;
  38.     DSU(int MaxNodes){
  39.         parent.resize(MaxNodes + 5);
  40.         for(int i = 0; i <= MaxNodes; i++) parent[i] = i;
  41.     }
  42.     int find_leader(int node){
  43.         return parent[node] = (parent[node] == node ? node : find_leader(parent[node]));
  44.     }
  45.     bool same_set(int u, int v){
  46.         return find_leader(u) == find_leader(v);
  47.     }
  48.     void union_sets(int u, int v){
  49.         int leader_u = find_leader(u), leader_v = find_leader(v);
  50.         if(leader_u ==leader_v) return;
  51.         parent[leader_u] = leader_v;
  52.     }
  53. };
  54.  
  55. void freq_fact(int n, int idx, vector< vector < int > >& adj){
  56.   for(int i = 1; i < sqrt(n); i++){
  57.     if(n % i == 0) adj[i].push_back(idx), adj[n / i].push_back(idx);
  58.   }
  59.   if(sqrt(n) == (int) sqrt(n)) adj[sqrt(n)].push_back(idx);
  60. }
  61.  
  62. int main(){
  63.   AhMeD_HoSSaM();
  64.   int test = 1;
  65.   TC {
  66.     int n;
  67.     cin >> n;
  68.     vector < int > nums(n);
  69.     vector < vector < int > > adj(1e5 + 1);
  70.     for(int i = 0; i < n; i++){
  71.         cin >> nums[i];
  72.         freq_fact(nums[i], i, adj);
  73.     }
  74.     ll compatibility = 0;
  75.     DSU dsu(1e5);
  76.     for(int i = 1e5; i >= 0; i--){
  77.       if(sz(adj[i]) < 2) continue;
  78.       int u = adj[i][0];
  79.       for(int j = 1; j < sz(adj[i]); j++){
  80.         int v = adj[i][j];
  81.         if(!dsu.same_set(u, v)) compatibility += i, dsu.union_sets(u, v);
  82.       }
  83.     }
  84.     cout << "Case " << test++ << ": " << compatibility << "\n";
  85.   }
  86.   Time
  87.   return 0;
  88. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement