等差、等比数列的类实现(Python)

等差、等比数列的类实现

  • 前言
    • 为何要写
  • 一、具体代码
  • 二、使用步骤
    • 1.等差数列
    • 2.等比数列
    • 3.重复读取的优化
  • 总结


前言

大家好,我又来了( ̄▽ ̄)/

这次是为大家带来了等差数列与等比数列的具体实现代码ヾ(◍°∇°◍)ノ゙


为何要写

首先我要说一下,我是如何一步一步过来的( Ĭ ^ Ĭ )

在我想为我自制的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_En

你可能感兴趣的:(编程练习,python,列表,统计学)