Leetcode刷题Python203.移除链表元素

Posted 2022-06-08 Better Bench

1 题目

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val val 的节点，并返回 新的头节点 。

2 解析

（1）方法一
迭代法，依次直接对head链表的每个元素进行判断。遇到待删除元素直接删除。注意，需要考虑一种情况，如果链表是[7,7,7,7]，删除的元素是7，是不能直接用head.next = head.next.next的，因为head.next.next就是空了。所以需要提前判断，如果链表是空，或者全是待删除的元素，要返回。

（2）方法二

采用虚拟节点迭代删除的方法，就不需要判断链表是否为空。注意，返回的是虚拟节点之后的链表，即是dummy_node.next
（3）方法三
递归法，有三个部分，第一是递归终止条件，第二是递归方法，第三删除操作。

3 python实现

（1）方法一

class Solution:
    # 方法一：迭代法

    def removeElements(self, head: ListNode, val: int) -> ListNode:
        # 判断所有元素都要要删除的
        while head and head.val ==val:
            head =head.next
        # 判断当前列表是否为空，或者是全部元素都是要删除的。比如[7,7,7,7],删除7返回是空
        if not head:
            return 
        # 删除指针都要
        temp = head
        while temp.next:
            if temp.next.val ==val:
                temp.next = temp.next.next
            else:
                temp = temp.next
        return head

（2）方法二

'''
class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next
    # 方法二：迭代第二种写法
def removeElements(self, head: ListNode, val: int) -> ListNode:
     dummy_node = ListNode(-1,head)
     cur = dummy_node
     while cur.next:
         if cur.next.val == val:
             cur.next = cur.next.next
         else:
             cur = cur.next
     # 注意返回的是虚拟头结点之后的链表，返回的是链表的next
     return dummy_node.next

（3）方法三

# 方法三：递归
def removeElements(self, head: ListNode, val: int) -> ListNode:
     # 递归终止条件
     if not head:
         return 
     # 递归
     head.next = self.removeElements(head.next,val)
     # 删除元素操作
     if head.val  == val:
         head = head.next
     return head