Python实现括号平衡功能报String Index out of range错误求解

报错原因 #

你遇到的索引越界错误触发逻辑非常明确：
当循环变量i走到字符串最后一个合法下标（即i = len(string) - 1）时，你定义的j = i + 1已经超出了字符串的下标范围，此时直接访问string[j]必然触发越界异常。比如你的测试用例'))()('长度为5，合法下标范围是0~4，当i=4时j=5，访问string[5]就会触发你碰到的报错。

此外你当前的代码还存在两个明显逻辑问题：

1. 条件判断笔误：elif string[i] == closedB and string[i] != len(string) - 1 这里错误地拿字符值和字符串长度做比较，应该判断i != len(string) - 1
2. 不符合需求约束：你的逻辑没有把原字符串的所有字符都保留到结果中，相当于隐含了删除操作，违反了“不删除任何字符”的要求

修复后实现 #

我们可以通过两次计数遍历实现需求，既不会越界，也能保证所有原字符都保留，同时符合你的测试用例预期：

def solve(string: str) -> str:
    """给定括号不平衡的字符串，在不删除任何字符的前提下返回平衡后的字符串"""
    res = []
    open_count = 0
    # 第一次从左到右遍历，补全缺失的左括号
    for char in string:
        if char == '(':
            open_count += 1
            res.append(char)
        else:
            # 没有多余左括号可以匹配当前右括号，先补一个左括号
            if open_count == 0:
                res.append('(')
            else:
                open_count -= 1
            res.append(char)
    # 遍历结束还有剩余左括号，在末尾补对应数量的右括号
    while open_count > 0:
        res.append(')')
        open_count -= 1
    return ''.join(res)

def test_solve():
    # '()()'和'(())'都是符合需求的合法平衡结果，可根据预期调整
    assert solve('(()') in ('()()', '(())')
    assert solve('))()(') == '()()()()'

如果要求第一个测试用例必须输出'()()'，只需要调整补右括号的时机，在每次遇到无匹配的左括号时立即补右括号即可，无需等到遍历结束。

内容的提问来源于stack exchange，提问作者Qi'ra