如何使用递归函数操作列表？

Question

我正试图以一种不寻常的方式（至少对我而言）操纵给定的列表。

基本上，我有列表a（也是图1），它有第一个索引作为主体。现在，我想遍历其他索引，如果某个值与第一个索引中的一个匹配，我想将该索引的子列表插入到第一个索引中。

我不知道我是否足够清楚，但目标应该是列表b（也是图2）。我认为这里应该使用递归函数，但我不知道如何。大家觉得有可能吗？

原名单：

a = [[1,2,3],[2,5],[6,3],[10,5]]

预期输出：

b = [[1,2,[2,5,[10,5]],3,[6,3]]]

Answer 1

您可以使用字典来记录每个数字在哪里第一次出现，记录在哪个列表中找到它，以及在哪个索引处。如果然后找到具有已经遇到的值的列表，则可以在插入匹配列表的情况下对记录的列表进行变异。如果没有匹配项（第一个列表 [1,2,3] 就是这种情况），那么这个列表只是附加到结果中。

因为插入到列表中会影响其他插入点，建议先收集插入动作，然后倒序应用：

下面是代码：

def solve(a):
    dct = {}
    result = []
    insertions = []
    for lst in a:
        found = None
        for i, val in enumerate(lst):
            if val in dct:
                found = val
            else:
                dct[val] = [lst, i]
        if found is None:
            result.append(lst)
        else:
            insertions.append((*dct[found], lst))

    for target, i, lst in reversed(insertions):
        target.insert(i + 1, lst)
        
    return result


# Example run:
a = [[1,2,3],[2,5],[6,3],[10,5]]
print(solve(a))

输出：

[[1, 2, [2, 5, [10, 5]], 3, [6, 3]]]