【Leetcode刷题Python】203.移除链表元素

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

猜你喜欢

转载自blog.csdn.net/weixin_43935696/article/details/125154683