给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-linked-list-elements/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处
暴力遍历,注意边界条件。
/**
* Definition for singly-linked list.
* function ListNode(val, next) {
* this.val = (val===undefined ? 0 : val)
* this.next = (next===undefined ? null : next)
* }
*/
/**
* @param {ListNode} head
* @param {number} val
* @return {ListNode}
*/
var removeElements = function(head, val) {
if (!head) return null
while(head.val === val) {
head = head.next
if (!head) return null
}
if (!head.next) return head
let prev = head, current = head.next
while (current) {
while(current.val === val) {
prev.next = current.next
current = current.next
if (!current) return head
}
prev = prev.next
if (!prev) return head
current = prev.next
}
return head
};
头结点:可以添加一个虚拟头结点,这样就不用考虑头的问题。