Skip to content

移除链表元素

约 509 字大约 2 分钟

快慢指针递归后序

2025-02-25

203. 移除链表元素

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

示例 1:

img.png

输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]

示例 2:

输入:head = [], val = 1
输出:[]

示例 3:

输入:head = [7,7,7,7], val = 7
输出:[]

提示:

  • 列表中的节点数目在范围 [0, 10^4]
  • 1 <= Node.val <= 50
  • 0 <= val <= 50

递归

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) {
            return head;
        }
        
        // 后序遍历,先解决子链表,再解决当前节点。
        head.next = removeElements(head.next, val);
        return head.val == val ? head.next : head;
    }
}
  • 时间复杂度:O(n),其中 n 是链表的长度。递归过程中需要遍历链表一次。
  • 空间复杂度:O(n),其中 n 是链表的长度。空间复杂度主要取决于递归调用栈,最多不会超过 n 层。

快慢指针

快指针无条件前进,慢指针有条件前进到快指针。检查快指针指向的元素。

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        // 移除链表开头连续的目标值节点
        while (null != head && head.val == val) {
            head = head.next; // 更新头节点
        }
        
        // 初始化当前节点和前一个节点,均为头节点
        ListNode cur = head;
        // pre 相当于构造链表的指针
        ListNode pre = head;

        // 遍历链表,直到当前节点为空。根据快指针的值确定更新慢指针的next还是前进慢指针到快指针
        while (cur != null) {
            // 如果当前节点的值等于目标值
            if (cur.val == val) {
                // 前一个节点的next指向当前节点的next,相当于跳过了当前节点。重点。
                pre.next = cur.next;
            } else {
                // 否则,前一个节点移动到当前节点
                pre = cur;
            }
            // 当前节点移动到下一个节点
            cur = cur.next;
        }
        
        // 返回处理后的链表头节点
        return head;
    }
}