每日一题——Python实现PAT甲级1108 Finding Average（举一反三+思想解读+逐步优化）

最新推荐文章于 2025-04-10 11:46:52 发布

用哲学编程

最新推荐文章于 2025-04-10 11:46:52 发布

阅读量947

点赞数 25

分类专栏：每日一题文章标签： python 开发语言算法

本文链接：https://blog.youkuaiyun.com/weixin_44915521/article/details/139621150

版权

每日一题专栏收录该内容

83 篇文章

订阅专栏

一个认为一切根源都是“自己不够强”的INTJ

个人主页：用哲学编程-优快云博客
专栏：每日一题——举一反三
 Python编程学习
 Python内置函数

我的写法

import math
N=int(input())
nums=input().split()
legal_num=N
output=0.0
for i in range(N):
    if nums[i].isalpha() or nums[i].count('.')>1:
        print(f"ERROR: {nums[i]} is not a legal number")
        legal_num-=1
    elif float(nums[i])<-1000 or float(nums[i])>1000:
        print(f"ERROR: {nums[i]} is not a legal number")
        legal_num-=1
    elif nums[i].count('.')==1 and len(nums[i].partition('.')[2])>2:
        print(f"ERROR: {nums[i]} is not a legal number")
        legal_num-=1
    else:
        output+=float(nums[i])
if legal_num==0:
    print("The average of 0 numbers is Undefined")
elif legal_num==1:
    print(f"The average of 1 number is {output:.2f}")
else:
    output/=legal_num
    print(f"The average of {legal_num} numbers is {output:.2f}")

时间复杂度

代码的时间复杂度主要由以下部分决定：

读取输入：O(N)
遍历每一个输入并验证合法性：O(N)
打印输出信息：O(1)

因此，整体时间复杂度为：O(N)。

空间复杂度

代码的空间复杂度主要由以下部分决定：

存储输入数值字符串的列表 nums：O(N)
存储一些常量（如 legal_num 和 output）：O(1)

因此，整体空间复杂度为：O(N)。

总结

这段代码实现了对输入数值的合法性检测和平均值计算，逻辑清晰，操作简单。时间复杂度和空间复杂度均为 O(N)，适用于输入数据量较大的情况。需要注意的是，这段代码假设输入的初始 N 和数值字符串是有效的，未处理可能的输入异常（如非整数的 N）。

我要更强

考虑到这段代码的主要功能是读取并处理输入字符串来计算平均值，其时间复杂度和空间复杂度已经较为高效，但仍有一些优化和改进的空间。

优化点

合并条件检查：减少重复的条件检查。
减少冗余计算：避免重复调用某些函数。
异常处理优化：对于转换为浮点数的操作，可以通过异常捕获来简化条件判断。

优化后的代码

def is_legal_number(num):
    try:
        # 尝试将 num 转换为浮点数
        val = float(num)
        # 检查数值范围
        if val < -1000 or val > 1000:
            return False
        # 检查小数点后的位数
        if '.' in num and len(num.split('.')[1]) > 2:
            return False
        return True
    except ValueError:
        # 转换失败说明不是合法的数字
        return False

def main():
    import sys
    input = sys.stdin.read
    data = input().split()
    
    N = int(data[0])
    nums = data[1:N+1]
    
    legal_num = 0
    output = 0.0
    
    for num in nums:
        if is_legal_number(num):
            output += float(num)
            legal_num += 1
        else:
            print(f"ERROR: {num} is not a legal number")
    
    if legal_num == 0:
        print("The average of 0 numbers is Undefined")
    elif legal_num == 1:
        print(f"The average of 1 number is {output:.2f}")
    else:
        output /= legal_num
        print(f"The average of {legal_num} numbers is {output:.2f}")

if __name__ == "__main__":
    main()