2021“MINIEYE杯”中国大学生算法设计超级联赛（9）

题目链接

---

8 Integers Have Friends 2.0（数论）

描述：
给出 $n(2\le n\le 2\ast 10^5)$ 个互不相同的数 $a_i(1\le a_i\le 4\ast 10^{12})$。对于一个整数 $m\ge 2$，如果 $a_{c_1}=a_{c_2}=...=a_{c_k}(modm)$，那么这些元素是在一个集合的。问可能产生的集合中，元素最多的集合元素个数是多少？

思路：
①考虑 $m$ 取 $2$ 时，$n$ 个元素被划分成了两类，所以最后答案集合的大小至少是 $\lceil \frac{n}{2}\rceil$ 。
②若$a_i=a_j(modm)$，则 $a_i-a_j$ 是 $m$ 的倍数。
③最优解的 $m$ 一定包含质数。
对于③粗略证明：
假设 $a_i=a_j(modpq)$，即$a_{i}modpq=a_{j}modpq=r$,
那么有 $a_i=pq{k}_i+r,a_j=pq{k}_j+r$
即 $a_i=p(q{k}_i+\lfloor \frac{r}{p}\rfloor )+<r{>}_p,a_j=p(q{k}_j+\lfloor \frac{r}{p}\rfloor )+<r{>}_p$
所以 $a_i=a_j(modp)$。
也就是说模上一个合数在一个集合中的元素，模上这个数的最小质因子也是在一个集合中的，集合的大小只增不减。
证毕！
考虑随机取两个不同的下标索引 $i,j$，$a_i$与$a_j$都处在最优解的集合中的概率是不低于 $\frac{1}{4}$ 的。如果 $a_i$ 与 $a_j$ 都在最优解的集合中，令 $val=|a_i-a_j|$，模数 $m$ 一定是 $val$ 的因数，并且我们可以只考虑 $val$ 的质因数。所以可以$O(\sqrt{max{a}_i})$枚举 $val$ 的质因子 $m$，因为 $4\ast 10^{12}$ 内的数最多只有 $11$ 个不同的质因数，所以可以 $O(11n)$ 来计算取每个质因数作为模数，并且包含 $a_i$ 与 $a_j$ 的集合大小，然后更新答案即可。
重复上述操作 $K$ 次，时间复杂度是$O(K\sqrt{max{a}_i}+11Kn)$，并且没有找到最优解的概率是 $(\frac{3}{4}{)}^K$，当 $K$ 取 $40$ 时，这个概率约等于$10^{-5}$，已经非常小了。

代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

const int N = 200010, M = 2000010, K = 40;

int n;
ll a[N];
int v[M], prime[M], m;
mt19937 gen(time(0));

void primes(int n)
{
    for (int i = 2; i <= n; i++)
    {
        if (!v[i])
            v[i] = i, prime[++m] = i;
        for (int j = 1; j <= m; j++)
        {
            if (prime[j] > v[i] || prime[j] * i > n) break;
            v[prime[j] * i] = prime[j];
        }
    }
}

int calc(ll m, ll r)
{
    int ans = 0;
    for (int i = 1; i <= n; i++)
        ans += a[i] % m == r;
    return ans;
}

int main()
{
    primes(M - 5);
    int t;
    scanf("%d", &t);
    while (t--)
    {
        scanf("%d", &n);
        for (int i = 1; i <= n; i++) scanf("%lld", &a[i]);
        int ans = 1;
        for (int i = 1; i <= K; i++)
        {
            int x, y;
            while (1)
            {
                x = gen() % n + 1;
                y = gen() % n + 1;
                if (x != y) break;
            }
            ll val = abs(a[x] - a[y]);
            for (int j = 1; (ll)prime[j] * prime[j] <= val; j++)
            {
                if (val % prime[j] == 0)
                {
                    ans = max(ans, calc(prime[j], a[x] % prime[j]));
                    while (val % prime[j] == 0)
                        val /= prime[j];
                }
            }
            if (val > 1)
                ans = max(ans, calc(val, a[x] % val));
        }
        printf("%lld\n", ans);
    }
    return 0;
}