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