CF1881D

题目大意

给你一个包含 $n$ 个正整数的数组 $a$，你可以进行以下操作：

1. 选两个数$a_i$和$a_j$ （$1＜i,j\le n$且$i\ne j$）
2. 选择$a_i$的一个因数$x$
3. 将$a_i$替换为$\frac{a_i}{x}$，$a_j$替换为$a_j\cdot x$

请判断是否能通过有限次操作使 $a$ 中所有元素相同（可以是$0$次）。

解题思路

我们可以发现无论做多少次操作· ${\prod }_{i=1}^n{a}_i$ 的值都是不变的。
所以我们可以得出如果可以通过操作，使 $a$ 中所有数相等，那么
$$\sqrt[n]{\prod _{i=1}^n{a}_i}$$
必然属于 $\mathbb{Z}$ (整数集)，可是题目里 $a_i\le 10^6$，$n\le 10^4$，最终答案可能达到 ${a_i}^{60000}$，所以我们可以给每个数分解质因数，判断每个因子的个数是否能被 $n$ 整除。

#include <bits/stdc++.h>

using namespace std;

int n, T, a[1000005];
map <int, int> mp;

void change(int x){
  for(int i = 2; i * i <= x; i ++){
    while(x % i == 0){
      mp[i] ++;
      x /= i;
    }
  }
  if(x > 1){
    mp[x] ++;
  }
  return ;
}

int main(){
  cin >> T;
  while(T --){
    mp.clear();
    cin >> n;
    for(int i = 1; i <= n; i ++){
      cin >> a[i];
      change(a[i]);
    }
    int flag = true;
    for(auto x : mp){
      if(x.second % n > 0){
        flag = false;
        break;
      }
    }
    if(flag == true){
      cout << "YES\n";
    }else{
      cout << "NO\n";
    }
  }
}