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

本文介绍了如何使用Python实现等差数列和等比数列,并通过类优化了重复读取的效率,降低了时间复杂度,适用于数学计算和数据分析场景。
摘要由CSDN通过智能技术生成


前言

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

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


为何要写

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

在我想为我自制的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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值