Weird copying linked list-ish thing with random

This commit is contained in:
2023-10-27 17:33:22 -04:00
parent 5df7489f27
commit efb8f453ae
2 changed files with 51 additions and 0 deletions
+26
View File
@@ -509,3 +509,29 @@ def count_factorial_trailing_zeroes(number: int) -> int:
divisor *= 5
return zeroes_count
def copy_random_list(
head: stdlib.ListNodeRandom | None,
) -> stdlib.ListNodeRandom | None:
if head is None:
return None
ordered = []
cur = head
while cur is not None:
ordered.append(cur)
cur = cur.next
ordered_copy = [stdlib.ListNodeRandom(entry.val) for entry in ordered]
hash_idx = {hash(n): i for i, n in enumerate(ordered)}
for i, entry in enumerate(ordered):
if i + 1 < len(ordered_copy):
ordered_copy[i].next = ordered_copy[i + 1]
if entry.random is not None:
ordered_copy[i].random = ordered_copy[hash_idx[hash(entry.random)]]
return ordered_copy[0]