OpenSoure_Repos/TheAlgorithms-Python
1
0
Fork 0
mirror of https://hub.njuu.cf/TheAlgorithms/Python.git synced 2023-10-11 13:06:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove extra len calls in doubly-linked-list's methods (#8600)

Browse Source
This commit is contained in:
amirsoroush 2023-04-01 15:23:21 +03:00 committed by GitHub
parent 3d2012c4ba
commit 63710883c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
data_structures/linked_list/doubly_linked_list.py
View File

@ -81,7 +81,9 @@ class DoublyLinkedList:
....
IndexError: list index out of range
"""
if not 0 <= index <= len(self):
length = len(self)
if not 0 <= index <= length:
raise IndexError("list index out of range")
new_node = Node(data)
if self.head is None:
@ -90,7 +92,7 @@ class DoublyLinkedList:
self.head.previous = new_node
new_node.next = self.head
self.head = new_node
elif index == len(self):
elif index == length:
self.tail.next = new_node
new_node.previous = self.tail
self.tail = new_node
@ -131,15 +133,17 @@ class DoublyLinkedList:
....
IndexError: list index out of range
"""
if not 0 <= index <= len(self) - 1:
length = len(self)
if not 0 <= index <= length - 1:
raise IndexError("list index out of range")
delete_node = self.head # default first node
if len(self) == 1:
if length == 1:
self.head = self.tail = None
elif index == 0:
self.head = self.head.next
self.head.previous = None
elif index == len(self) - 1:
elif index == length - 1:
delete_node = self.tail
self.tail = self.tail.previous
self.tail.next = None