蓝桥杯-等差数列（python，动态规划）

一、题目

蓝桥杯2019年第十届省赛真题-等差数列
时间限制: 1Sec 内存限制: 128MB 提交: 7338 解决: 1666
题目描述
数学老师给小明出了一道等差数列求和的题目。但是粗心的小明忘记了一 部分的数列，只记得其中 N 个整数。

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

输入
输入的第一行包含一个整数 N。 第二行包含N个整数A1,A2,···,AN。(注意A1 ∼AN并不一定是按等差数

列中的顺序给出)

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

输出
输出一个整数表示答案

样例输入
5
2 6 4 10 20
样例输出
10

二、思路和代码

思路：动态规划，注释很详细。

代码：

import math

N  = int(input())
ls = [int(x) for x in input().split()]
ls.sort()
# 让数据下标为1-N
ls.insert(0, 0)

# 动态规划，用dp[i]记录前i项可能的最大的差。
dp = [0 for i in range(N+1)]
dp[2] = ls[2] - ls[1]

# 状态转换方程为dp[i] = (ls[i]-ls[i-1])和dp[i-1]的最大公因数
for i in range(3, N+1):
    dp[i] = math.gcd(ls[i]-ls[i-1], dp[i-1])

if dp[N] == 0:
    print(N)
else:
    print((ls[-1]-ls[1])//dp[N]+1)