移除链表元素

[louzhedong]

习题

出处 LeetCode 算法第203题

删除链表中等于给定值 val 的所有节点。

示例:

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

思路

顺序遍历

解答

/**
 * @param {ListNode} head
 * @param {number} val
 * @return {ListNode}
 */
var removeElements = function(head, val) {
    var first = new ListNode();
    first.next = head;
    
    var flag = first;
    
    while(head) {
        if (head.val == val) {
            first.next = head.next;
            head = head.next;
        } else {
            head = head.next;
            first = first.next;
        }
    }
    
    return flag.next;
};