Python开发arithmetic_arranger函数实现算术式位数校验求助

实现步骤说明 #

首先修正现有代码的两处问题：

1. 原运算符校验逻辑错误：if "+" not in problem or "-" not in problem 会把所有合法输入都判定为错误，只要运算式只包含+或者只包含-就会触发报错，需要先修正这个逻辑。
2. 原代码用print输出错误不符合要求，需求是返回错误字符串，统一换成return返回结果。

补充位数校验的逻辑很简单：把每个运算式按运算符拆成左右两个操作数字符串，直接判断字符串长度是否超过4即可，不需要转整数，开销更低。

完整补全后的代码 #

import re

def arithmetic_arranger(problems):
    # 校验问题数量
    if len(problems) > 5: 
        return "Error: Too many problems."
    
    for problem in problems: 
        # 校验运算符类型
        if '+' in problem:
            op = '+'
        elif '-' in problem:
            op = '-'
        else:
            return "Error: Operator must be '+' or '-'."
        
        # 拆分并清洗左右操作数
        left_num, right_num = problem.split(op)
        left_num = left_num.strip()
        right_num = right_num.strip()
        
        # 校验操作数仅含数字
        if re.search('[^0-9]', left_num) or re.search('[^0-9]', right_num):
            return "Error: Numbers must only contain digits."
        
        # 新增：校验操作数位数不超过4位
        if len(left_num) > 4 or len(right_num) > 4:
            return "Error: Numbers cannot be more than four digits."
    
    # 所有校验通过，可在此处补充后续运算式排列的逻辑
    return "All validations passed"

测试验证 #

• 调用arithmetic_arranger(["32 + 698", "3801 - 2", "45 + 43", "123 + 49"])返回All validations passed，符合正常通过要求
• 调用arithmetic_arranger(["32 + 698", "38011 - 2", "45 + 43", "123 + 49"])返回指定的位数错误字符串，符合报错要求

内容的提问来源于stack exchange，提问作者ranelk