Daimayuan Online Judge #555. 整齐的数组2

算法分析

随机化 + 小学数学 ???

两个数之间若能通过 k 变成一样的数 说明 abs (a[i] - a[j]) % k == 0 即 k 是 abs (a[i] - a[j]) 的因子 那么随机取两个数 a[i] 和 a[j],枚举 abs(a[i] - a[j])的因子作为 k ,判断 k 是否合法,更新答案即可

那么随机化如何保证正确性呢 n 个数中至少有 1 / 2 的数属于答案的集合.那么随机取一个数,取到答案数的概率至少为 1 / 2,取出两个数倘若均不为答案的可能性为 3 / 4 那么 (3 / 4) ^ 1000 是非常小的 因此此做法可行.

AC Code

#include<bits/stdc++.h>
using namespace std;
const int N = 100;
int a[N];
int n; 

int check(int k)
{
    map<int,int> mp;
    for(int i = 1;i <= n;i ++)//统计 n 个数 mod k 的余数 
    {
        mp[(a[i] % k + k) % k]++;
    }
    for(auto i : mp)//倘若已经有超过一半的数 mod k 余数相同 那么方案成立
    {
        if(i.second*2>=n)return 1;
    }
    return 0;
}

void solve()
{
    scanf("%d",&n);
    for(int i = 1;i <= n;i ++) scanf("%d",&a[i]);
    map<int,int> mp;
    for(int i = 1;i <= n;i ++) mp[a[i]] += 1;

    int f = 0;
    for(auto it : mp)
    {
        if(it.second * 2 >= n) f = 1;
    }
    if(f) {puts("-1"); return ;}

    srand(time(NULL));
    int res = 0;
    for(int R = 1;R <= 1000;R ++)//随机 1000 轮
    {
        int rd1 = rand() % n + 1;//取出俩随机数
        int rd2 = rand() % n + 1;
        if(rd1 == rd2) continue;
        int x = abs(a[rd1] - a[rd2]);
        for(int j = 1;j * j <= x;j ++)//枚举其因子
        {
            if(x % j == 0) 
            {
                if(check(j)) res = max(res , j);
                if(check(x / j)) res = max(res , x / j);
            }
        }
    }
    printf("%d\n",res);
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T --)
    {
       solve();
    }
    return 0;
}