LeetCode 203:移除链表元素 Remove LinkedList Elements

删除链表中等于给定值 val 的全部节点。java

Remove all elements from a linked list of integers that have value val.python

示例:函数

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

解题思路:

两种方法,一种是迭代法,从第一个节点开始,遇到值相同的节点就将其删除。链表的删除操做是直接将删除节点的前一个节点指向删除节点的后一个节点便可。学习

第二种方法是递归,用递归从后向前遇到相同节点直接指向该节点的下一个节点的地址便可指针

迭代法:

因为链表删除操做的特殊性,若是要删除某个节点,必需要知道该删除节点的前一个节点地址才可完成删除操做。因此若是是从第一个节点开始判断,就要考虑到第一个节点是否为空节点、第一个节点是否就是该删除的的节点,删除头节点和非头节点的操做不同,应单独实现删除操做。code

若是原链表是这种形式:1->1->1->2 val=1 删除头节点后第二个节点置为头节点,可是第二个节点做为新的头节点依然须要删除,因此对头节点的操做应该是一个迭代过程。递归

另一种方法就是新建一个虚拟节点,该虚拟节点下一个节点指向原链表头节点。这就无需考虑头节点是否为空、是否为待删除节点。如原链表为:1->1->1->2 val=1 ,新建一个虚拟节点 -1 做为头节点:-1->1->1->1->2 val=1 ,这时只需正常迭代删除便可,惟一要注意是返回节点不能是 head ,由于原链表的头节点若是是待删除节点,此时 head 节点已被删除并做为单独隔离出来的节点,并不是链表内的一个节点。element

Java:rem

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode newHead=new ListNode(-1);//新建虚拟节点
        newHead.next=head;//虚拟节点做为原链表的头节点
        ListNode cur = newHead;//遍历节点的指针

        while (cur.next != null) {
            if (cur.next.val == val) {
                cur.next = cur.next.next;//删除操做
            } else {
                cur = cur.next;
            }
        }
        return newHead.next;//返回的头节点应当是虚拟节点的下一个节点
    }
}

Python3:io

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        newHead = ListNode(-1)
        newHead.next = head
        cur = newHead
        while cur.next:
            if cur.next.val == val:
                cur.next = cur.next.next
            else:
                cur = cur.next
        return newHead.next

递归法:

递归方法解该题很简单,基线条件是遇到空节点(最后一个节点),递归时只需将传递参数节点的下一个节点做为新的参数传给递归函数便可:

如原链表为: 1->2->6->3->4->5->6, val = 6 递归到空节点时最后一个递归函数返回null: 1->2->6->3->4->5->6->null 回到上一层递归函数内此时 head.next 获得返回节点 null 判断head为 6 删除,返回 head.next :1->2->6->3->4->5->null 回到上一层递归函数此时 head.next 获得上一层返回节点依然为null 判断head为 5 不删除,返回 head 为5:1->2->6->3->4->5->null 回到上一层递归函数此时 head.next 获得上一层返回节点 5 判断head为 4 不删除,返回 head 为4:1->2->6->3->4->5->null 回到上一层递归函数此时 head.next 获得上一层返回节点 4 ..............

直到回到第一个递归函数为止返回头节点结束。

Java:

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) return null;//基线条件
        head.next = removeElements(head.next, val);
        if (head.val == val) {
            return head.next;
        } else {
            return head;
        }
    }
}

Python3:

class Solution:
    def removeElements(self, head: ListNode, val: int) -> ListNode:
        if not head: return None
        head.next = self.removeElements(head.next, val)
        if head.val == val:
            return head.next
        else:
            return head

欢迎关注公众号一块儿学习:爱写Bug

爱写Bug.png

相关文章
相关标签/搜索