题目大意
有一个正整数序列
a
1
,
⋯
,
a
n
a_1,\cdots,a_n
a1,⋯,an,每次操作可以把一个数
+
1
+1
+1 或
−
1
-1
−1,但要使其仍为正数。问至少多少次操作,使得整个序列的
gcd
\gcd
gcd 不为
1
1
1。
n
≤
2
×
1
0
5
n \le 2 \times 10^5
n≤2×105
2.5s
\\
\\
\\
题解
首先有些很明显的观察,比如,可以枚举一个
a
i
a_i
ai 的质因数
p
p
p 作为
gcd
\gcd
gcd,然后每个数变成
min
(
a
i
m
o
d
p
,
p
−
a
i
m
o
d
p
)
\min(a_i \bmod p,p-a_i \bmod p)
min(aimodp,p−aimodp),这样判一次是
O
(
n
)
O(n)
O(n) 的。
再比如,答案不会超过序列中奇数的数量,因为每个奇数
+
1
+1
+1 就会使得
2
∣
gcd
2|\gcd
2∣gcd。
然后。。。
就没有然后了。。。
然后题解说,虽然判一个质因子要
O
(
n
)
O(n)
O(n),但是只用判很少的质因子。
由观察 2,答案不超过
n
n
n,因此最终改动在
1
1
1 以内的位置至少有
n
2
\frac n2
2n 个。也就是说,任选一个位置,至少有
n
2
\frac n2
2n 的概率,最终的
gcd
\gcd
gcd 被它的质因子整除。
因此就随机选序列的 20 个数,判断
a
i
,
a
i
+
1
,
a
i
−
1
a_i,a_i+1,a_i-1
ai,ai+1,ai−1 的所有质因子,就行了~
代码
#include<bits/stdc++.h>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long LL;
const int maxn=2e5+5;
int n;
LL a[maxn];
LL _check(LL p)
{
LL ans=0;
fo(i,1,n) ans+=(a[i]<p) ?p-a[i] :min(a[i]%p,p-a[i]%p);
return ans;
}
LL check(LL x)
{
LL ans=n+500;
int sqrtx=sqrt(x);
for(int i=2; i<=sqrtx; i++) if (x%i==0)
{
ans=min(ans,_check(i));
while (x%i==0) x/=i;
}
if (x>1) ans=min(ans,_check(x));
return ans;
}
int main()
{
scanf("%d",&n);
fo(i,1,n) scanf("%lld",&a[i]);
LL ans=0;
fo(i,1,n) ans+=(a[i]&1);
random_shuffle(a+1,a+1+n);
random_shuffle(a+1,a+1+n);
random_shuffle(a+1,a+1+n);
fo(i,1,min(n,20))
{
ans=min(ans,check(a[i]-1));
ans=min(ans,check(a[i]));
ans=min(ans,check(a[i]+1));
}
printf("%lld\n",ans);
}