如何像递增数字一样递增指定字符集的字符串？求Python代码片段

这个需求本质是基于自定义字符集的进制数递增，每个字符对应进制中的一位，从右向左处理进位逻辑即可实现。以下是对应的Python代码：

charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890"
# 构建字符与索引的双向映射，提升查询效率
char_to_idx = {char: idx for idx, char in enumerate(charset)}
idx_to_char = {idx: char for idx, char in enumerate(charset)}
charset_length = len(charset)

def increment_custom_string(s):
    # 转成列表方便修改单个字符（字符串不可变）
    char_list = list(s)
    # 从最后一位开始向前遍历
    ptr = len(char_list) - 1
    
    while ptr >= 0:
        current_idx = char_to_idx[char_list[ptr]]
        if current_idx < charset_length - 1:
            # 当前字符不是字符集最后一位，直接递增
            char_list[ptr] = idx_to_char[current_idx + 1]
            return ''.join(char_list)
        else:
            # 当前字符是最后一位，重置为字符集首位，继续向前进位
            char_list[ptr] = idx_to_char[0]
            ptr -= 1
    
    # 如果所有字符都触发进位，在开头添加字符集首位
    return idx_to_char[0] + ''.join(char_list)

# 测试示例
print(increment_custom_string("aabbcde"))  # 输出: aabbcdf
print(increment_custom_string("aabbcd0"))  # 输出: aabbcea

代码说明 #

• 双向映射字典：char_to_idx和idx_to_char避免了重复调用charset.index()，大幅提升字符查找的效率，尤其适合字符集较大的场景。
• 进位逻辑：从字符串末尾开始检查，能直接递增则修改后返回；若当前字符是字符集最后一位，则重置为首位并向前一位继续处理进位。
• 边界处理：当所有字符都处于字符集末尾时（比如输入"000"），会自动在字符串开头添加字符集的第一个字符，返回"a000"。

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