数学老师给小明出了一道等差数列求和的题目。
但是粗心的小明忘记了一部分的数列,只记得其中 N 个整数。
现在给出这 N 个整数,小明想知道包含这 N 个整数的最短的等差数列有几项?
输入格式
输入的第一行包含一个整数 N。
第二行包含 N 个整数 A1,A2,⋅⋅⋅,AN。(注意 A1∼AN 并不一定是按等差数
列中的顺序给出)
输出格式
输出一个整数表示答案。
数据范围
2
≤
N
≤
100000
,
2≤N≤100000,
2≤N≤100000,
0
≤
A
i
≤
1
0
9
0≤Ai≤10^9
0≤Ai≤109
输入样例:
5
2
5\quad2
52
6
4
10
20
6\quad4\quad10\quad20
641020
输出样例:
10
10
10
样例解释
包含 2、6、4、10、20 的最短的等差数列是 2、4、6、8、10、12、14、16、18、20。
分析
假设数列{An},按照顺序排列后,后一项减前一项一定是最小公差的整数倍。即 α 1 d , α 2 d , α 3 d . . . α n d \alpha_1d,\;\alpha_2d,\;\alpha_3d\;...\;\alpha_nd α1d,α2d,α3d...αnd
问题就变成了求新数列的最大公因数。
注意特判公差为0的情况
参考代码
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 100010;
int a[N], d[N];
int n;
//求最大公因数,欧几里德算法
int gcd(int a, int b){
return b ? gcd(b, a % b) : a;
}
int main (){
cin >> n;
for(int i = 0;i < n;i ++) cin >> a[i];
sort(a, a + n);
//后一项减前一项构造出新的数列
for(int i = 1;i < n;i ++) d[i - 1] = a[i] - a[i - 1];
int t;
//求数列的最大公因数
t = gcd(d[0], d[1]);
for(int i = 2;i < n - 1;i ++) t = gcd(d[i], t);
if(a[0] == a[n - 1]) cout << n << endl; //特殊情况特判!
else cout << (a[n - 1] - a[0]) / t + 1;
return 0;
}