Codeforces Round #694 (Div. 2) D. Strange Definition（平方数在质因数分解下的意义）

传送门

---

题目大意

给出一个序列，定义序列中的两个数$a,b$是$adjacent$当且仅当$\frac{lcm(a,b)}{gcd(a,b)}$是平方数。现在每秒会发生如下操作，设序列中的每个数$a_i$和序列中的如下$j$个数 { b 1 , b 2 , . . . , b j } \{b_1,b_2,...,b_j\} {b1​,b2​,...,bj​}为$adjacent$（包括它自己），那么$a_i$会被替换为${\prod }_{k=1}^j{b}_k$。设每个数对应的$b$序列的大小为$d_i$，那么这一秒的答案为 m a x { d i } max\{d_i\} max{di​}，给出$q$个询问，每次询问输出$w$秒时的答案。

解题思路

简单打表可以发现，实际上第一秒到以后任意秒的答案都是不变的。但是如何找到第$0$秒时每个数的$d_i$呢。

首先将$\frac{lcm(a,b)}{gcd(a,b)}$化为$\frac{a\ast b}{gcd(a,b{)}^2}$，我们发现分母是平方数，也就是说分母质因数分解后的每个质因子的幂次都为偶数，那么问题就转化为了什么情况下$a\ast b$也是平方数：

若两个数$x,y$相乘为平方数，设对$x$质因数分解为$p_1^{a_1}{p}_2^{a_2}...p_n^{a_n}$，对$y$质因数分解为$q_1^{b_1}{q}_2^{b_2}...q_n^{b_n}$。对于某个公共的质数$p$在$x,y$中的幂次分别为$a,b$，那么一定有$a+b$为偶数，即$a\equiv b(mod2)$。那么问题变成了，对于序列中的任何$x$，转化为$p_1^{a_1\%2}{p}_2^{a_2\%2}...p_n^{a_n\%2}$，这样转化后相等的两个数若相等即乘积为平方数。

对于序列中的所有数，要么转化为$1$要么转化为若干个质数相乘。然后看每个数有多少个，如果有奇数个那么会一直保持不变，因为奇数个数相乘对应的幂次仍为奇数，偶数的话第0秒后就变成了$1$，当然还要特判一下初始即为1的数。

#include <bits/stdc++.h>

using namespace std;
#define ENDL "\n"
typedef long long ll;
const int maxn = 1e6 + 10;

int minp[maxn];
vector<int> prime;
unordered_map<int, int> mp;

void init() {
    for (int i = 1; i < maxn; i++) minp[i] = i;
    for (int i = 2; i < maxn; i++) {
        if (minp[i] == i) prime.push_back(i);
        for (int j = 0; j < prime.size() && 1LL * i * prime[j] < maxn; j++) {
            minp[i * prime[j]] = prime[j];
            if (i % prime[j] == 0) break;
        }
    }
}

int divide(int n) {
    map<int, int> fac;
    while (n > 1) {
        fac[minp[n]]++;
        n /= minp[n];
    }
    int ans = 1;
    for (auto i : fac) {
        if (i.second & 1) ans *= i.first;
    }
    return ans;
}

int main() {
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int t, n, q;
    ll w;
    init();
    cin >> t;
    while (t--) {
        cin >> n;
        mp.clear();
        for (int i = 1, x; i <= n; i++) {
            cin >> x;
            int y = divide(x);
            mp[y]++;
        }
        int spj = 0, odd = 0;
        for (auto i : mp) {
            spj = max(i.second, spj);
            if (i.first == 1) {
                odd += i.second;
                continue;
            }
            if (!(i.second & 1)) {
                odd += i.second;
            }
        }
        //cout << odd << " " << spj << ENDL;
        cin >> q;
        while (q--) {
            cin >> w;
            if (w == 0) {
                cout << spj << ENDL;
            } else {
                cout << max(spj, odd) << ENDL;
            }
        }
    }
    return 0;
}