递归函数无限循环排查：匹配初始元素时终止函数需求

原代码的问题是没设置「递归回到初始元素即终止」的条件，当B中存在指向初始子列表第一个元素的项时，递归会无限循环。下面是针对性的修改方案：

修改要点 #

• 给递归函数map_multiple新增root_element参数，用来记录每个子列表的初始元素（也就是终止节点）
• 在匹配到符合条件的value时，先判断是否等于root_element：
  • 若是，直接生成当前路径并返回，不再继续递归
  • 若不是，再执行原有的递归逻辑
• 调用函数时，传入每个子列表的第一个元素作为root_element

修改后的完整代码 #

existing_list = [[ '10000714484,10000714481']]

A = df_test['A']
B = df_test['B']


def map_multiple(sublist, visited, root_element, max_depth=10):
    subsets = []
    anyMatches = False
    for (key, value) in zip(A, B):
        if key == sublist[-1]:
            # 匹配到初始元素，终止递归，直接返回当前路径+该元素
            if value == root_element:
                subsets.append(sublist + [value])
                anyMatches = True
            # 未匹配到初始元素且未访问过，继续递归
            elif value not in visited:
                new_visited = set(visited)
                new_visited.add(value)
                new_sets = map_multiple(sublist + [value], new_visited, root_element)
                subsets.extend(new_sets)
                anyMatches = True
    if not anyMatches:
        return [sublist]
    return subsets


new_list = []
for i in range(len(existing_list)):
    sublist = existing_list[i]
    # 传入子列表的第一个元素作为root_element
    allSubsets = map_multiple(sublist, set(sublist), sublist[0])
    new_list.extend(allSubsets)

# 去重并转换格式
new_list = list(set(map(tuple, new_list)))
new_list = [list(x) for x in new_list]

print(new_list)

关键逻辑说明 #

• 新增的root_element参数固定了每个递归链的终止节点，确保一旦遍历回到初始元素就停止递归
• 把「匹配到初始元素」的判断放在最前面，优先处理终止逻辑，避免进入无限递归
• 保留了原有的visited集合，防止路径中出现重复元素导致的循环

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