Python字符串末尾数字整体加1功能实现代码纠错求助

你的代码逻辑有两个核心bug，导致多场景运行不符合预期：

• 只判断了字符串最后1位是不是数字，没有提取末尾完整的连续数字段，自然没法把整段数字当整体做加法
• 用str.replace()做替换是全局匹配替换，只要字符串里出现和最后一位相同的数字，所有匹配项都会被替换，比如输入foobar11时，最后一位是1，replace会把字符串里两个1都换成2，就输出了错误的foobar22；遇到末尾是9需要进位的场景（比如foobar99），单字符替换完全处理不了进位逻辑
• 代码最后一行return(strng)是无效代码，if和else分支里已经写了return，这行永远不会被执行

不用记复杂的高级语法，按三步拆解逻辑就行：

1. 拆分字符串：从字符串最后一位往前遍历，直到碰到非数字字符就停止，把字符串拆成「前面非数字前缀」和「末尾连续数字段」两部分
2. 分支处理：
   • 如果末尾没有数字段，直接在前缀后面拼接1就是结果
   • 如果有数字段，先把数字段转成整数加1，再把加完的结果转回字符串，补前导零到和原数字段相同的长度（如果加1后进位导致长度超过原数字段，比如99变100，直接用进位后的结果就行，不用截断）
3. 把前缀和处理好的新数字段拼接起来，就是最终结果

def increment_string(strng):
    # 初始默认数字段起始位置在字符串末尾，即无数字段
    num_start_index = len(strng)
    # 从后往前遍历找连续数字的起点
    current_index = len(strng) - 1
    while current_index >= 0 and strng[current_index].isdigit():
        num_start_index = current_index
        current_index -= 1
    
    # 切分前缀和数字段
    prefix = strng[:num_start_index]
    num_part = strng[num_start_index:]

    # 无末尾数字的情况
    if not num_part:
        return prefix + "1"
    
    # 处理数字加1和前导零
    origin_num_length = len(num_part)
    new_num = int(num_part) + 1
    # zfill方法会自动在字符串前补0到指定长度，超出长度时不做截断
    new_num_str = str(new_num).zfill(origin_num_length)

    return prefix + new_num_str

• 输入foobar → 输出foobar1，符合预期
• 输入foobar001 → 输出foobar002，符合预期
• 输入foobar11 → 输出foobar12，不会出现全局替换导致的foobar22错误
• 输入foobar99 → 输出foobar100，正确处理进位逻辑

内容的提问来源于stack exchange，提问作者Bożena