初阶数据结构习题【3】（1时间和空间复杂度）——203移除链表元素

1. 题目描述

力扣在线OJ——移除链表元素
给你一个链表的头节点 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
输出：[ ]

2. 思路

思路1：找到等于 val 的节点直接删除

3 代码实现

struct ListNode* removeElements(struct ListNode* head, int val) {
    struct ListNode* cur = head;
    struct ListNode* prev = NULL;
    while (cur) {
        if (cur->val != val) {
            prev = cur;
            cur = cur->next;
        } else {
            if (prev == NULL) {
                head = cur->next;
                free(cur);
                cur = head;
            } else {
                prev->next = cur->next;
                free(cur);
                cur = prev->next;
            }
        }
    }

    return head;
}