蓝桥杯——等差数列

原题链接：等差数列

题目描述

数学老师给小明出了一道等差数列求和的题目。

但是粗心的小明忘记了一部分的数列，只记得其中 $N$ 个整数。

现在给出这 $N$ 个整数，小明想知道包含这 $N$ 个整数的最短的等差数列有几项？

输入格式

输入的第一行包含一个整数 $N。$

第二行包含 $N$ 个整数 $A_1,A_2,\cdot \cdot \cdot ,A_N。$(注意 $A_1\sim A_N$ 并不一定是按等差数列中的顺序给出)

输出格式

输出一个整数表示答案。

数据范围

$2\le N\le 100000,$
$0\le Ai\le 109$

输入样例1：

5
2 6 4 10 20

输出样例1：

10

样例解释

包含 $2、6、4、10、20$ 的最短的等差数列是 $2、4、6、8、10、12、14、16、18、20$

思路

等差数列项数的公式为 $(a_n-a_1)/d+1$，其中 $a_n$ 为等差数列的末项，在序列中就是最大值，$a_1$ 为等差数列的首项，在序列中就是最小值

所以要求序列最短，因为最大值和最小值已经固定，要求的就是公差 $d$ 的最大值

因为等差序列可以表示为 $a_1,a_1+d,a_1+2d,a_1+3d,....,a_1+nd$ ，可以观察到等差数列的每一项和第一项的差值都是 $d$ 的倍数，所以求 $d$ 的最大值就是求序列给出的项数差值的最大公约数

时间复杂度

$O(nlogn)$

C++ 代码

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 100010;
int a[N];
int n;

int gcd(int a,int b) {
    return b ? gcd(b, a % b) : a;
}

int main()
{
    scanf("%d",&n);
    for(int i = 0;i < n;i ++) scanf("%d",&a[i]);
    sort(a,a + n);
    
    int d = 0;
    for(int i = 1;i < n;i ++) d = gcd(d, a[i] - a[i - 1]);
    
    if(!d) printf("%d\n",n);
    else printf("%d\n",(a[n - 1] - a[0]) / d + 1);
    
    return 0;
}

---