LeetCode.203-移除链表元素

209 阅读1分钟

给你一个链表的头节点 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

解题思路

依次遍历链表,如果当前节点的下一个节点的值等于指定值,那么就将当前节点的下一个节点指定为下下个节点。这里我们要注意一点如果第一个节点的值为指定值的话就不好返回了,所以我们定义一个返回节点指向头节点,然后再定义一个指针指向返回节点去进行遍历。

/**
 * 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) {
  let ret = new ListNode(null)
  ret.next = head
  let cur = ret
  while(cur.next) {
    if (cur.next.val === val) {
      cur.next = cur.next.next
    } else {
      cur = cur.next
    }
  }
  return ret.next
};