P8682 [蓝桥杯 2019 省 B] 等差数列 Python

蓝桥杯 2019 省 B 等差数列

题目描述

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

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

输入格式

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

第二行包含 N N N 个整数 A 1 , A 2 , ⋯   , A N A_1,A_2,\cdots,A_N A1,A2,⋯,AN。(注意 A 1 ∼ A N A_1 ∼ A_N A1∼AN 并不一定是按等差数列中的顺序给出 )。

输出格式

输出一个整数表示答案。

样例 #1

样例输入 #1

复制代码
5
2 6 4 10 20

样例输出 #1

复制代码
10

提示

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

对于所有评测用例, 2 ≤ N ≤ 1 0 5 2 \le N \le 10^5 2≤N≤105, 0 ≤ A i ≤ 1 0 9 0 \le A_i \le 10^9 0≤Ai≤109。

蓝桥杯 2019 年省赛 B 组 H 题。

思路

因为给定数据是无序的,若是想知道数据之间的联系,肯定是要排序的。

最开始的想法是求除相邻数的差,最小值即为公差。但这种思路是错误的,比如 【2 6 12 20 30】 最小值是4,但公差应为2 所以我们继续考虑这些差值有什么联系

【4,6,8,10】 他们的最大公因数应为2 ,所以差值的最大公因数应该是公差。

在本代码中,并没有对所有相邻数的差值求最大公因数,只选取了0,1,和最后一个数,求差值的公差d,所以单靠该条件无法判断公差。

所以用差值的最小值num_d与d比较,若相等,则用等差数列公式求n,若不等,则用最小值求。

另外需要注意的是,题目要求长度最短,所以在给定数据中,最大值和最小值即为首项和尾项;若num_d和d = 0时,需要特殊讨论,公差为0,比如【2,2,2,2,2】;获取的n的值,即为等差数列的项数。

python 复制代码
import math  
n = int(input())
a = list(map(int,input().split()))
a.sort()
num_d = math.inf
# print("num_d is",num_d)
if len(a)==2:
    num = 2
else:   
    for i in range(1,len(a)-1):
        if num_d > a[i] - a[i-1]:
            num_d = a[i] - a[i-1]
            
    d = math.gcd(a[-1]-a[0],a[1]-a[0])
    if num_d == d and d!=0:
        num = (a[-1] -a[0]) /d +1
    elif num_d == d==0:
        num = n
    else:
        num = (a[-1] -a[0]) /num_d + 1
# print(d)
print(int(num))
相关推荐
linweidong2 分钟前
中科闻歌Java面试题及参考答案
java·python·大模型·提示词·ai agent·mcp·rag原理
小玮看世界11 分钟前
[Python]OD算法在OD实际运用转化参考清单
开发语言·python·算法
萧鼎14 分钟前
2026实战:用 tomlkit 优雅读写 TOML 配置,告别手写解析器
python·开源·教程·python库
小刘在重生~1 小时前
Java 常用类|包装类 装箱拆箱、Integer 缓存面试题
java·python·缓存
2601_962099461 小时前
Python xlwt设置excel单元格字体及格式
python·excel·xlwt·样式设置·单元格格式
奇牙coding1231 小时前
GPT-6-Astra API 接入教程:OpenRouter 路由配置 + Python/curl 示例 + 静默降级踩坑
开发语言·python·gpt·ai
2601_962298272 小时前
Python与Selenium结合的Web自动化测试全流程实践教程
自动化测试·python·selenium·web测试·pageobjectmodel
天衍四九-2 小时前
第一章:从 LLM 到 Agent —— DeepSeek Harness 入门
网络·数据库·人工智能·python
529宝宝起名网2 小时前
用 Python 实现名字寓意评分算法:基于 NLP 语义分析的名字内涵深度评估
python·算法·自然语言处理
2601_962298673 小时前
Python:对象缓存优化机制(CPython)
python·性能优化·内存优化·cpython·对象缓存