Rekursive Funktion zum Rückgängigmachen einer beliebig verschachtelten Liste

Question

Ich versuche, eine "einfache" rekursive Funktion zu schreiben, um einen Baum zu durchlaufen, der durch geschachtelte Listen repräsentiert wird. Ich habe die richtige Ausgabe, wenn ich print verwende, aber ich möchte das Ergebnis in einer list zurückgegeben werden.

-Code

def trav(t): 
    while len(t) > 1: 
     for counter, i in enumerate(t): 
      if isinstance(i, list): 
       t2 = t.pop(counter) 
       trav(t2) 
    n=t.pop() 
    print(n) 

Run es

tree1 = [1, [2, [4], [5]], [3]] 
trav(tree1) 

Ausgabe über Druck:

4 
5 
2 
3 
1 

Wunsch Ausgabe über Rückgabewert:

[4, 5, 2, 3, 1] 

Answer 1

Deklarieren Sie einen Akkumulator acc, aggregieren Sie den Rückgabewert des rekursiven Aufrufs in der Schleife.

Am Ende Rückkehr acc + der letzte popped Wert (dies fügt sich gut in den Basisfall).

def trav(t): 
    acc = [] 
    while len(t) > 1: 
     for counter, i in enumerate(t):   
      if isinstance(i, list): 
       t2 = t.pop(counter) 
       acc += trav(t2) 

    n = t.pop() 
    return acc + [n] 



tree1 = [1, [2, [4], [5]], [3]] 
print(trav(tree1)) 
# [4, 5, 2, 3, 1]