蓝桥杯 等差数列 python组

题目描述

数学老师给小明出了一道等差数列求和的题目。但是粗心的小明忘记了一 部分的数列，只记得其中 N 个整数。

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

输入描述

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

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

其中，2≤N≤10^5，0≤Ai​≤10^9。

解答：

因为A1到An不一定是按等差数列的顺序输出的，所以在输入之后要先进行排序，因为没有哪两个值确定是相邻的，即公差 不一定等于所有数中的最小差值，所以要计算“最大公约数”，求出最小的最大公约数，然后将（（最大值 - 最小值）//公差 ） +1 ，就是包含这n个整数的最短的等差数列个项数。

import math    #调用了math库中的gcd函数，即求最大公约数的方法

n = int(input())
ans = list(map(int,input().split()))

ans.sort()
d = ans[-1]-ans[0]
if d==0:
  print(len(ans))    #公差为0，就是所有的值全部都一样，所以最短的项数就是输入的数值的个数
else:
  for i in range(2,len(ans)):    #遍历求出所有相邻数的差值，和当前d的最大公约数，求出满足输入的n个数为等差数列的公差的值
    d = math.gcd(d,ans[i]-ans[i-1])
  print(((ans[-1]-ans[0])//d)+1)    #输出最小的等差数列的项数