题目链接:707. 设计链表 - 力扣(LeetCode)
题目描述:
实现 MyLinkedList
类:
MyLinkedList()
初始化MyLinkedList
对象。int get(int index)
获取链表中下标为index
的节点的值。如果下标无效,则返回-1
。void addAtHead(int val)
将一个值为val
的节点插入到链表中第一个元素之前。在插入完成后,新节点会成为链表的第一个节点。void addAtTail(int val)
将一个值为val
的节点追加到链表中作为链表的最后一个元素。void addAtIndex(int index, int val)
将一个值为val
的节点插入到链表中下标为index
的节点之前。如果index
等于链表的长度,那么该节点会被追加到链表的末尾。如果index
比长度更大,该节点将 不会插入 到链表中。void deleteAtIndex(int index)
如果下标有效,则删除链表中下标为index
的节点。
代码:
class ListNode:
def __init__(self,val = 0,next = None):
self.val = val
self.next = next
class MyLinkedList:
def __init__(self):
self.dummy_head = ListNode()
self.size = 0
def add(self,val:int)->None:
new_node = ListNode(val)
cur = self.dummy_head
while cur.next:
cur = cur.next
cur.next = new_node
self.size += 1
def get(self,index:int)->int:
if index < 0 or index >= self.size:
return -1
cur = self.dummy_head.next
for i in range(index):
cur = cur.next
return cur.val
obj = MyLinkedList()
obj.add(0)
obj.add(4)
obj.add(2)
obj.add(3)
print(obj.get(1)) # 4