小鑫的算法之路：leetcode0203 移除链表元素

题目

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

解法1：迭代法

针对节点删除，为了操作的归一，需要新创建一个哑结点，并指向该头结点。然后遍历链表的节点，如果该节点的值刚好等于val，那么直接删除该节点即可。

代码如下：

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* dummy = new ListNode(0, head);
        ListNode* pre = dummy;
        while (pre->next != nullptr) {
            if (pre->next->val != val) {
                pre = pre->next;
            } else {
                ListNode* cur = pre->next;
                pre->next = cur->next;
                delete cur;
            }
        } 
        ListNode* ans = dummy->next;
        delete dummy;
        return ans;
    }
};

时间复杂度为O(n)，空间复杂度为O(1)。测试结果如下：

解法2：递归法

本题也可通过递归法处理，如果本节点不为空且指向的值刚好等于val，则需要删除本节点，并且返回下一个节点值。当然如果本节点不等于val，则只需在下一个节点满足条件的场景下更新本节点的next指针。递归终止条件为节点为nullptr。

代码如下：

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        if (head == nullptr) {
            return nullptr;
        }
        
        head->next = removeElements(head->next, val);

        if (head->val == val) {
            ListNode* next = head->next;
            delete head;
            return next;
        }

        return head;
    }
};

时间复杂度为O(n)，空间复杂度为O(n)。测试结果如下：