问题描述
数学老师给小明出了一道等差数列求和的题目。但是粗心的小明忘记了一部分的数列,只记得其中 N 个整数。
现在给出这 N 个整数,小明想知道包含这 N 个整数的最短的等差数列有几项?
输入格式
输入的第一行包含一个整数 N。
第二行包含 N 个整数 A₁, A₂, · · · , AN。(注意 A₁ ∼ AN 并不一定是按等差数列中的顺序给出)
输出格式
输出一个整数表示答案。
样例输入
5
2 6 4 10 20
样例输出
10
样例说明
包含 2、6、4、10、20 的最短的等差数列是 2、4、6、8、10、12、14、16、 18、20。
思路:
1.由输入可得得首项和末项
2.公差为排序后各前后两项只差中的最小值
3.求得公差、首项、末项后可直接求出数列长度
注意:
特殊值:当输入N个同一个数字时,公差为零,输出N
N=int(input())
li=list(map(int,input().split()))
li.sort()
an=[]
d=0
for i in range(1,len(li)):
an.append(li[i]-li[i-1])
an.sort()
d=an[0]
if d==0:
print(N)
else:
print(int(1+(li[len(li)-1]-li[0])/d))
总结:核心算法是“公差为排序后各前后两项之差中的最小值”,难度简单。