LeetCode-203. 移除链表元素

98 阅读2分钟

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

示例 1:

removelinked-list.jpg

输入: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, 104] 内
1 <= Node.val <= 50
0 <= val <= 50

解题思路

1.逆递归 我们可以通过逆递归的方式先递归的最后一个节点,然后最后开始判断当前节点的值是否为目标值,如果是将当前节点的下一个节点返回,如果不是就返回当节点。然后我再把每一次递归返回的值都赋值给上一个节点的next。

/**
 * @param {ListNode} head
 * @param {number} val
 * @return {ListNode}
 */
var removeElements = function(head, val) {
    // 链表题老规矩,判断第一个节点是否是空节点
    if (!head) return head
    // 逆递归
    head.next = removeElements(head.next, val)
    return head.val === val ? head.next : head 
};

2.迭代 我们先定义一个指针指向当前头节点,然后从头节点开始迭代整个链表。判断当前节点的下一个节点的值是否是目标值,如果是那么就把当前节点的下一个节点设置为下下个节点,如果不是那就指针指向下一个节点。同时我们需要定义一个最顶级的节点,让这个最顶级的节点的下一个节点指向最开始头节点,这样我们最后只用返回这个顶级节点的下一个节点了。

/**
 * @param {ListNode} head
 * @param {number} val
 * @return {ListNode}
 */
var removeElements = function(head, val) {
    if (!head) return head
    // 迭代
    let retHead = new ListNode(0)
    retHead.next = head
    let cur = retHead
    while (cur.next) {
        if (cur.next.val === val) {
            cur.next = cur.next.next
        } else {
            cur = cur.next
        }
    }
    return retHead.next
};