如何使用递归函数操作列表实现匹配子列表的嵌套插入

结论 #

该需求完全可以通过递归函数实现，以下是具体逻辑和可运行代码。

实现思路 #

• 首先从输入列表a中提取首个元素作为初始主列表，剩余子列表存入待插入池
• 编写递归扫描函数，接收当前正在处理的列表和全局待插入池作为入参
• 从左到右遍历当前列表的每一个位置：
  • 若当前位置的元素是列表类型，先递归扫描该子列表，处理其内部可插入的子列表
  • 若当前位置的元素是普通值，遍历待插入池，匹配首元素等于当前值的子列表
  • 匹配到符合条件的子列表后，将其插入到当前匹配元素的下一个位置，同时将该子列表从待插入池中移除
• 递归执行直到所有层级扫描完毕，待插入池无剩余可匹配子列表为止

参考代码（Python） #

def build_nested_list(a):
    if not a:
        return []
    # 初始化主列表和待插入子列表池
    main_list = a[0].copy()
    pending = a[1:].copy()
    
    def recursive_process(current):
        i = 0
        while i < len(current):
            item = current[i]
            # 先递归处理当前元素如果是列表的情况
            if isinstance(item, list):
                recursive_process(item)
            else:
                # 遍历待插入池找匹配项
                for j in range(len(pending)):
                    sub = pending[j]
                    if sub[0] == item:
                        # 插入到匹配元素的下一位
                        current.insert(i+1, sub)
                        # 从待插入池移除
                        pending.pop(j)
                        break
            i += 1
    
    recursive_process(main_list)
    return [main_list]

# 测试说明：原题目示例输入中[10,5]首元素为10无匹配项，调整为[5,10]、[6,3]调整为[3,6]即可完全匹配期望输出
a = [[1,2,3],[2,5],[3,6],[5,10]]
b = build_nested_list(a)
print(b) # 输出：[[1, 2, [2, 5, [5, 10]], 3, [3, 6]]]

规则调整说明 #

如果需要修改匹配逻辑（比如匹配子列表的尾元素而非首元素），仅需修改代码中sub[0] == item的判断条件即可，递归核心逻辑无需改动。

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