初阶数据结构习题【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* prev = NULL;
    struct ListNode* cur = head;
    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;
}