Solution -「CF 1622F」Quadratic Set

$\mathscr{Description}$

  Link.

  求 $S\subseteq\{1,2,\dots,n\}$，使得 $\prod_{i\in S}i$ 是完全平方数，并最大化 $|S|$。

  $n\le10^6$。

$\mathscr{Solution}$

  爆搜打出 $20$ 以内的表，发现 $|S|\approx n$。先研究偶数 $n=2k$：

$$\begin{aligned} \prod_{i=1}^{2k} i! &= \left( \prod_{i=1}^k i! \right)^2 \prod_{i=1}^k 2i\\ &= \left( \prod_{i=1}^k i! \right)^2 2^k k!. \end{aligned}$$

那么若 $2^kk!$ 是完全平方数，有 $|S|=n$；否则若 $2^k$ 是完全平方数，有 $|S|=n-1$，删去 $k!$ 即可；否则至少有 $|S|=n-2$，只需要删去 $2!$ 和 $k!$。继而，对于奇数 $n=2k+1$，答案至少为 $n-3$。

  所以，我们只需要判断 $|S|$ 能否取 $n,n-1,n-2$。这里有个 trick：异或哈希。对于每个素数 $p$，独立随机生成 hash 值 $h(p)$，并定义 $h(ab)=h(a)\oplus h(b)$，这样就能对每个数建立 hash，若两数 $x,y$ 的唯一分解中指数奇偶性完全一致，就应有 $h(x)=h(y)$。利用这个 trick，求出所有 $h(i!)$ 后顶多拿 unordered map 判一判就能完成 $S$ 的取值判断了。复杂度为 $\mathcal O(n)$（假定 std::unordered_map 为 $\mathcal O(1)$ 操作）。

$\mathscr{Code}$​

/*+Rainybunny+*/

#include <bits/stdc++.h>

#define rep(i, l, r) for (int i = l, rep##i = r; i <= rep##i; ++i)
#define per(i, r, l) for (int i = r, per##i = l; i >= per##i; --i)

typedef unsigned long long ULL;

const int MAXN = 1e6;
int n, pn, pr[MAXN + 5];
bool npr[MAXN + 5];
ULL hnum[MAXN + 5], hfac[MAXN + 5];
std::unordered_map<ULL, int> buc;

inline void init() {
    std::mt19937_64 emt(time(0) ^ 20120712);
    rep (i, 2, n) {
        if (!npr[i]) hnum[pr[++pn] = i] = emt();
        for (int j = 1, t; j <= pn && (t = i * pr[j]) <= n; ++j) {
            npr[t] = true, hnum[t] = hnum[i] ^ hnum[pr[j]];
            if (!(i % pr[j])) break;
        }
    }
    rep (i, 1, n) hfac[i] = hfac[i - 1] ^ hnum[i];
}

int main() {
    scanf("%d", &n), init();

    ULL h = 0;
    rep (i, 1, n) h ^= hfac[i];
    
    if (!h) {
        printf("%d\n", n);
        rep (i, 1, n) printf("%d%c", i, i < n ? ' ' : '\n');
        return 0;
    }

    rep (i, 1, n) if (h == hfac[i]) {
        printf("%d\n", n - 1);
        rep (j, 1, n) if (i != j) printf("%d ", j);
        return putchar('\n'), 0;
    }

    rep (i, 1, n) buc[hfac[i]] = i;
    rep (i, 1, n) if (buc.count(h ^ hfac[i])) {
        printf("%d\n", n - 2); int tmp = buc[h ^ hfac[i]];
        rep (j, 1, n) if (j != i && j != tmp) printf("%d ", j);
        return putchar('\n'), 0;
    }

    printf("%d\n", n - 3);
    rep (i, 1, n - 1) if (i != 2 && i != n >> 1) printf("%d ", i);
    return putchar('\n'), 0;
}