2019年第十届蓝桥杯 - 省赛 - C/C++大学B组 - H. 等差数列
·
【问题描述】
数学老师给小明出了一道等差数列求和的题目。
但是粗心的小明忘记了一部分的数列,只记得其中 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。
Ideas
因为 A1 ∼ AN 并不一定是按等差数列中的顺序给出,所以要先排一下序。
然后既然这是又给等差数列,那么我们当然首先要求出第 i 项和第 i + 1 项的差,构成一个 diff 数组。
之后我们再求出 diff 数组中所有数的最大公约数,这个最大公约数就是包含这 N 个整数的最短的等差数列的公差。
然后就是等差数列的内容了,我们用最后一项减去第一项,然后用差除以上面算出来的公差,得到的结果就是这个最短的等差数列中一共有多少个公差,然后再加上1,就是等差数列的长度。
最后要考虑一下边界情况,如果 diff 数组中所有数的最大公约数等于 0 呢,那就说明数组中存在相等的元素,此时我们直接返回 N 就可以了。
Code
Python
def gcd(a, b):
return gcd(b, a % b) if b else a
if __name__ == '__main__':
n = int(input())
nums = list(map(int, input().split(' ')))
nums.sort()
diff = [nums[i] - nums[i - 1] for i in range(1, n)]
ans = gcd(diff[0], diff[1])
for i in range(2, len(diff)):
ans = gcd(ans, diff[i])
print(((nums[-1] - nums[0]) // ans) + 1)
更多推荐


所有评论(0)