leetcode203. 移除链表元素

Posted 2021-01-25 xinfenglee

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

示例:

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

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* removeElements(struct ListNode* head, int val) {
    while(head && head->val == val)
    {
        struct ListNode* tmp = head;
        head = head->next;
        free(tmp);
    }
    if(head == NULL) return head;
    struct ListNode* cur = head;
    while(cur->next)
    {
        if(cur->next->val == val)
        {
            struct ListNode* tmp = cur->next;
            cur->next = cur->next->next;  
            free(tmp);
        }
        else
        {
            cur = cur->next;
        }
    }
    return head;
}