给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
示例 1:
输入: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
};