203. 移除链表元素

题目链接

https://leetcode.cn/problems/remove-linked-list-elements/description/

解题思路

按照我们解决递归的思路,我们首先想,这个递归函数,应该返回什么,应该定义什么参数。

显而易见,递归要解决什么问题,就要返回什么参数。即,我们要返回一个无val元素的链表。

而这个递归函数需要一个结点当作参数。

然后我们去想,本层需要做什么,出口是什么。

显然,如果本层结点等于val,那val结点直接就不要了,返回递归函数处理过的head.next即可。

如果本层结点不等于val,那保留本层结点,把递归函数处理过的链表(即它的返回值)接到本层结点后面即可。

那退出条件(出口)呢?这个链表走到了尽头(None的情况)

代码

class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next
class Solution:
    def removeElements(self, head, val: int):
        if head is None:
            return None
        if head.val == val:
            return self.removeElements(head.next, val)
        else:
            head.next = self.removeElements(head.next, val)
        return head

 二刷:

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        if not head:
            return None
        if head.val == val:
            return self.removeElements(head.next, val)
        head.next = self.removeElements(head.next, val)
        return head

 

posted @ 2023-01-03 10:29  BJFU-VTH  阅读(28)  评论(0编辑  收藏  举报