等差数列（求最短数列长度）最大公约数解法

等差数列（求最短数列长度）最大公约数解法

问题描述

数学老师给小明出了一道等差数列求和的题目。但是粗心的小明忘记了一部分的数列，只记得其中 N 个整数。现在给出这 N 个整数，小明想知道包含这 N 个整数的最短的等差数列有几项？

输入格式

输入的第一行包含一个整数 N。
第二行包含 N 个整数 A1; A2; · · · ; AN。 (注意 A1 ∼ AN 并不一定是按等差数列中的顺序给出)

输出格式

输出一个整数表示答案。

样例输入

5
2 6 4 10 20

样例输出

10

样例说明

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

评测用例规模与约定

对于所有评测用例， 2 ≤ N ≤ 100000， 0 ≤ Ai ≤ 109

思路

使用最大的公差去递增，可以达到最短长度

1. 对输入的数进行排序
2. 求出每相邻两个数的公差
3. 求出2中所有公差的最大公约数，结果就是等差数列的公差
4. 用公差、首项和尾项求出等差数列的项数

求出两数的最大公约数

//LL为long long int
//递归方法
LL gcd(LL a, LL b) {
    return (b == 0) ? (a) : (gcd(b, a % b));
}

完整代码

//
// Created by Charon on 2022/1/23.
//

#include "algorithm"
#include "iostream"

using namespace std;

#define MAX 100000

typedef long long LL;


LL gcd(LL a, LL b) {
    return (b == 0) ? (a) : (gcd(b, a % b));
}

int main() {
    LL N;
    cin >> N;

    LL a[MAX], delta[MAX]; //delta为各个数的公差
    for (int i = 0; i < N; ++i) {
        cin >> a[i];
    }

    sort(a, a + N); //排序

    //求公差
    for (int i = 0; i < N - 1; ++i) {
        delta[i] = a[i + 1] - a[i];
    }

    //求所有公差的最大公约数
    LL max = gcd(delta[0], delta[1]);
    for (int i = 1; i < N - 2; ++i) {
        LL gcdn = gcd(max, delta[i + 1]);
        if (max > gcdn) {
            max = gcdn;
        }
    }

    if (max == 0)
        cout << N << endl;
    else
        cout << (a[N - 1] - a[0]) / max + 1 << endl;

    return 0;
}