203. Remove Linked List Elements 未看到时间结果

Remove all elements from a linked list of integers that have value val.
Example*****Given:* 1 --> 2 --> 6 --> 3 --> 4 --> 5 --> 6, val = 6
Return: 1 --> 2 --> 3 --> 4 --> 5

这个的思路很简单,从头开始遍历,遍历到不是的就删了,这里要注意头节点就不是的情况。
我使用了一个假节点和两个指针:

var removeElements = function(head, val) {
    if (!head)
        return head;
    var pre = new ListNode(1);
    pre.next = head;
    var pointer = head;
    var headFlag = false;
    while(pointer) {
        if (pointer.val===val) 
            pre.next = pointer.next;
        else {
            if (!headFlag) {
                head = pointer;
                headFlag = true;
            }
            pre = pointer;
        }
        pointer = pointer.next;
    }
    if (!headFlag) 
        return null;
    return head;
};

在讨论里看到了更好的办法,使用一个假节点和一个指针,思想是一样的,但是这个明显更好。

var removeElements = function(head, val) {
    if(head===null) return head;
    var fake = new ListNode(0);
    fake.next = head;
    var cur = fake;
    while(cur.next !== null){
        if(cur.next.val === val){
            cur.next = cur.next.next;
        }else{
            cur = cur.next;
        }
    }
    return fake.next;
};

你可能感兴趣的:(203. Remove Linked List Elements 未看到时间结果)