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

Update remove_kth_node.py

Browse Source
This commit is contained in:
marina 2023-08-07 16:17:11 -07:00 committed by GitHub
parent d8d1887a85
commit 0389271f6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
linked_lists/remove_kth_node.py
View File

@ -7,25 +7,24 @@ def remove_kth_node(self, head, n):
if head is None or head.next is None:
return None
node = head
lenght = 0
# find the lenght of the list
# find the length of the list
node, length = head, 0
while node:
node = node.next
lenght += 1
if n == lenght:
length += 1
# if n is the entire list, remove head
if n == length:
return head.next
i = 0
node = head
while i < lenght - n - 1:
# loop to kth element
node, i = head, 0
while i <= length - n:
node = node.next
i += 1
# remove the kth element
node.next = node.next.next
return head