LeetCode 203：移除链表元素 python

LeetCode 203：移除链表元素 python

第一次用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))

四、结果