前言
大家好,我又来了( ̄▽ ̄)/
这次是为大家带来了等差数列与等比数列的具体实现代码ヾ(◍°∇°◍)ノ゙
为何要写
首先我要说一下,我是如何一步一步过来的( Ĭ ^ Ĭ )
在我想为我自制的mAth模块添加计算[a, b]区间内,间隔为n的数之和的函数的时候[・_・?]
首先在我脑海里浮现的,当然就是用for循环建立一个符合要求的列表,然后运用python的内建函数sum()
(*´゚∀゚`)ノ
但这样的话,该函数的时间复杂度为O(n) ╮(╯﹏╰)╭。
所以,我便寻找有没有时间复杂度更低的方法。
就这样,我把目光放在了我初中学的等差数列ヾ(゚∀゚ゞ)。
这样的话就可以将时间复杂度从O(n)降到了O(1),果然前人的智慧很强大呀(゚▽゚*)
至于等比数列则是连带的实现了。
前言少徐,让我们书归正传(✪ω✪过了一把评书瘾)
一、具体代码
让我们话不多说,直接列出本章需要的所有代码
from enum import Enum, unique
@unique
class Arithmetic_Progression_Enum(Enum):
Sn = 0
an = 1
class Geometric_Progression_Enum(Enum):
Sn = 0
an = 1
def Arithmetic_Progression(a1, d, n, result_flag):
if result_flag == Arithmetic_Progression_Enum.Sn:
Sn = n * a1 + (n * d * (n - 1)) / 2
if round(Sn) == Sn:
return round(Sn)
else:
return Sn
elif result_flag == Arithmetic_Progression_Enum.an:
an = a1 + (n - 1) * d
return an
def Arithmetic_Progression_List(a1, d, start, end):
lSt = []
for temp_n in range(start, end+1):
lSt.append(Arithmetic_Progression(a1, d, temp_n, result_flag=Arithmetic_Progression_Enum.an))
return lSt
def Geometric_Progression(a1, q, n, result_f