Git-Mirrors/master-algorithms-py
1
0
Fork 0
mirror of https://github.com/autistic-symposium/master-algorithms-py.git synced 2025-04-29 20:26:07 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update flatten_list.py

Browse Source
This commit is contained in:
marina 2023-08-07 16:22:14 -07:00 committed by GitHub
parent ea12ccce74
commit 85f1f87c8d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
linked_lists/flatten_list.py
View File

@ -27,14 +27,15 @@ def dfs(prev, node):
return dfs(last, temp_next) return dfs(last, temp_next)
def flatten(head): def flatten(head):
if head is None: if head is None:
return head return head
pseudo_head = Node(None, None, head, None) sentinel = Node(None, None, head, None)
dfs(pseudo_head, head) dfs(prev=sentinel, node=head)
pseudo_head.next.prev = None
return pseudo_head.next # erase the pointer to sentinel and return
sentinel.next.prev = None
return sentinel.next