优化基于stack、queue与func的迭代处理流程效率问题

问题背景 #

我拥有一个stack（元素数组的数组，初始形式为[[element]]）、一个queue（element数组），以及一个将(element1, element2)映射为resulting_element的函数func。原本采用以下代码处理：

for new_element in queue:                                   #first loop
    stack_copy = [list(array) for array in stack]
    stack[0].append(new_element)
    for (index, array) in enumerate(stack_copy):            #second loop
        for element in array:                               #third loop
            resulting_element = func(element, new_element)
            try:
                stack[index+1].append(resulting_element)
            except:
                stack.append(list())
                stack[index+1].append(resulting_element)

当前实现通过创建stack_copy遍历修改前的栈状态，但这种拷贝整个栈数组的方式效率极低。需求是：将new_element插入stack[0]，并将resulting_element插入stack[index+1]（位置无关），希望优化处理速度，考虑将stack改写为可在迭代时修改的生成器，且保持两个内循环的结构不变。

优化方案 #

核心思路是避免创建完整的栈拷贝列表，改用生成器惰性迭代修改前的栈状态，同时确保迭代过程不受后续栈修改的影响。具体修改如下：

for new_element in queue:
    # 用生成器预存修改前的每个数组（转成元组避免后续修改影响）
    prev_stack_generator = (tuple(array) for array in stack)
    # 先插入新元素到stack[0]
    stack[0].append(new_element)
    # 保持原内循环结构不变，遍历生成器
    for index, array in enumerate(prev_stack_generator):
        for element in array:
            resulting_element = func(element, new_element)
            try:
                stack[index+1].append(resulting_element)
            except IndexError:
                # 直接创建新列表并添加元素，简化异常处理
                stack.append([resulting_element])

优化点说明 #

• 替换栈拷贝为生成器：使用生成器表达式代替列表推导式创建拷贝，无需一次性生成整个拷贝列表，节省内存开销，惰性求值的特性也能提升迭代效率。
• 用元组固定数组状态：将每个数组转为元组，避免后续修改stack[0]时影响迭代的数组内容，确保遍历的是修改前的栈元素，和原逻辑完全一致。
• 简化异常处理：将通用except改为捕获IndexError，避免意外捕获其他异常，同时直接用stack.append([resulting_element])代替先创空列表再添加的冗余操作。

如果你的数组元素都是不可变类型，甚至可以直接迭代原栈数组（无需转元组），但转元组能确保在元素可变的场景下也不会出现逻辑错误，兼容性更强。

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