20210605 每日一题 移除链表元素

题目

题目链接

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {

    }
};

方法一：迭代

分析

迭代方法是解决链表问题的最简单的方法，用 $temp$ 表示当前节点，题目要求删除链表中所有节点值等于特定值的节点，可以将条件转化为：
$$temp=\{\begin{array}{ll}temp.next& iftemp.val=specialvalue\\ temp& iftemp.val\ne specialvalue\end{array}$$ 由于链表操作当前节点容易出错，所以一般操作链表下一节点，即 $temp.next$ ，将上述公式转化为：
$$temp.next=\{\begin{array}{ll}temp.next.next& iftemp.next.val=specialvalue\\ temp.next& iftemp.next.val\ne specialvalue\end{array}$$ 当 $temp.next=null$ 时，链表遍历结束。
特殊情况： 由于操作的是 $temp.next$，无法实现方面对链表的头节点 $head$ 的删除操作，因此需要在当前头节点 $head$ 前连接一个虚拟头节点 $virtualHead$，从虚拟头节点开始遍历链表，最后返回 $virtualHead$。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* virtualHead = new ListNode(0, head);
        ListNode* temp = virtualHead;
        while(temp->next != nullptr) {
            if(temp->next->val == val) {
                temp->next = temp->next->next;
            } else {
                temp = temp->next;
            }
        }
        return virtualHead->next;
    }
};

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是链表的长度，迭代需要遍历链表每个节点。
• 空间复杂度：$O(1)$。

方法二：递归

分析

链表题目一般都可以用递归方法来解，先递归到达链表末端进行处理，再将处理完的链表逐级返回链表前端。可以将条件转化为：
$$\{\begin{array}{ll}returnnode.next& ifnode.val=specialvalue\\ returnnode& ifnode.val\ne specialvalue\end{array}$$

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        if(head == nullptr) return head;
        head->next = removeElements(head->next, val);
        return head->val == val ? head->next : head;
    }
};

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是链表的长度。
• 空间复杂度：$O(n)$，空间复杂度主要取决于递归调用栈，最多不会超过 n 层。

拓展