Advertisement
Guest User

Untitled

a guest
Mar 31st, 2020
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.35 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3.  
  4. #pragma GCC optimize("Ofast,no-stack-protector")
  5. #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")
  6. #pragma GCC optimize("unroui-loops")
  7. #pragma GCC optimize("fast-math")
  8. #pragma GCC optimize("profile-values,profile-reorder-functions,tracer")
  9. #pragma GCC optimize("vpt")
  10. #pragma GCC optimize("rename-registers")
  11. #pragma GCC optimize("move-loop-invariants")
  12. #pragma GCC optimize("unswitch-loops")
  13. #pragma GCC optimize("function-sections")
  14. #pragma GCC optimize("data-sections")
  15. #pragma GCC optimize("branch-target-load-optimize")
  16. #pragma GCC optimize("branch-target-load-optimize2")
  17. #pragma GCC optimize("btr-bb-exclusive")
  18.  
  19. using namespace std;
  20.  
  21. int gcd(int a, int b) {
  22.     return b ? gcd(b, a % b) : a;
  23. }
  24.  
  25. void dfs(int v, int color, vector<int> &prev, vector<int> &used, vector<vector<int>> &adj, vector<vector<bool>> &matr) {
  26.     for (int u : adj[v]) {
  27.         if (used[u] == 0) {
  28.             bool f = false;
  29.             for (int g : prev) {
  30.                 if (!matr[g][u]) f = true;
  31.             }
  32.             if (f) continue;
  33.             prev.push_back(u);
  34.             used[u] = color;
  35.             dfs(u, color, prev, used, adj, matr);
  36.         }
  37.     }
  38. }
  39.  
  40. int main() {
  41.     ios::sync_with_stdio(false);
  42.     int t; cin >> t;
  43.     cout << gcd(437, 391) << endl;
  44.     for (int temp = 0; temp < t; temp++) {
  45.         int n; cin >> n;
  46.         vector<int> nums(n);
  47.         for (int i = 0; i < n; i++) cin >> nums[i];
  48.         vector<vector<int>> adj(n, vector<int>());
  49.         vector<vector<bool>> matr(n, vector<bool>(n, false));
  50.         for (int i = 0; i < n; i++) {
  51.             for (int j = i; j < n; j++) {
  52.                 if (gcd(nums[i], nums[j]) > 1) {
  53.                     adj[i].push_back(j);
  54.                     adj[j].push_back(i);
  55.                     matr[i][j] = matr[j][i] = true;
  56.                 }
  57.             }
  58.         }
  59.         vector<int> used(n, 0);
  60.         int color = 0;
  61.         for (int i = 0; i < n; i++) {
  62.             if (used[i] == 0) {
  63.                 color++;
  64.                 used[i] = color;
  65.                 vector<int> prev;
  66.                 dfs(i, color, prev, used, adj, matr);
  67.             }
  68.         }
  69.         cout << color << endl;
  70.         for (int i = 0; i < n; i++) {
  71.             cout << used[i] << ' ';
  72.         }
  73.         cout << endl;
  74.     }
  75. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement