第一次用python写链表,我在怀疑自己到底有没有学过python,
题目已经定义了节点,所以不用自己写,但是没给定义链表啊,呜呜呜所以调试用掉了我很大很大的力气
一、题目
题目传送门—— https://leetcode-cn.com/problems/remove-linked-list-elements/
二、解析
最简单的单链表删除元素,是链表模块的入门级别了,链表删除主要考虑三种情况
1.空链表,也就是head == None,这种情况直接不用匹配return None
p = head
if p == None:
return None
2.头节点元素 == val,删除头节点。
head.val ==val时,head = head.next头节点后移
注意:此步骤使用while可以覆盖多个头节点需要删除的情况
# 头节点删除
while (head and head.val == val):
head = head.next
3.非头节点元素 == val,删除节点 。
p = head q = p.next,当q.val == val时,p.next = q.next像是跳过了val那个节点 ,节点后移q=q.next
注意:此步骤使用while需要判断head != None的条件,可以覆盖全部元素均匹配到val时的情况不报错
# 非头节点删除
q = p.next
while (p.next != None):
if q.val == val:
p.next = q.next
q = q.next
else:
p = q
q = q.next
三、编码 1.定义节点
此节点类,leetcode 已给出,最好记住方便以后自己定义使用
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
2.定义链表
链表中两个功能函数,initList() 用于初始化链表,printList() 用于打印出链表
class LinkList:
def __init__(self):
self.head=None
def initList(self, data):
# 创建头结点
self.head = ListNode(data[0])
r=self.head
p = self.head
# 逐个为 data 内的数据创建结点, 建立链表
for i in data[1:]:
node = ListNode(i)
p.next = node
p = p.next
return r
def printlist(self,head):
if head == None:
return None
node = head
while node != None:
print(node.val,end=' ')
node = node.next
3.删除元素
class Solution:
def removeElements(self, head: ListNode, val: int) -> ListNode:
p = head
#判断链表是否空
if p == None:
return None
# 头节点删除
while (head and head.val == val):
head = head.next
# 非头节点删除
q = p.next
while (p.next != None):
if q.val == val:
p.next = q.next
q = q.next
else:
p = q
q = q.next
return head
4.调试验证
if __name__=='__main__':
solution = Solution()
linklist = LinkList()
d = [6,1,3,6]
data = linklist.initList(d)
print("处理前链表")
linklist.printlist(data)
print("\n处理后链表")
linklist.printlist(solution.removeElements(data,6))
四、结果
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)