DAY26 Remove Linked List Elements

第 11 屆 iThome 鐵人賽

DAY 26

自我挑戰組

刷題記錄與人生分享系列第 26 篇

11th鐵人賽 leetcode c javascript

keyhouse12

2019-09-27 08:35:29

1565 瀏覽

分享至

題目:

https://leetcode.com/problems/remove-linked-list-elements/
刪除鏈結串列中所有指定的數值。

解題思路:

一開始先用2個指標紀錄首節點和首節點的下一個，當先行的指標不等於要求的數值，把前一個指標連到現在指標然後移動現在指標至下一個指標，直到條件不符合為止。

C版本:

struct ListNode* removeElements(struct ListNode* head, int val) {
        while(head!=NULL && head->val == val)
            head = head->next;
        if(head == NULL)
            return NULL;
        struct ListNode * p = head->next;
        struct ListNode * q = head;
        while(p)
        {
            if(p->val != val)
            {
                q->next = p;
                q = p;
            }
            else
            {
                q->next = NULL;
            }
            p = p->next;
        }
        return head;
}

Javascript版本:

var removeElements = function(head, val) {
    while(head != null && head.val == val)
        head = head.next;
    
    if(head == null)
        return null;
    var post = head.next;
    var cur = head;
    
    while(post)
    {
        if(post.val != val)
        {
            cur.next = post;
            cur = post;
        }
        else
        {
            cur.next = null;
        }
        post = post.next;
    }
    return head;
};