leetcode203.移除链表元素

  参考网址：代码随想录

一、直接使用原来的链表进行移除节点操作

思路：移除头结点和移除其他节点的操作是不一样的，因为链表的其他节点都是通过前一个节点来移除当前节点，而头结点没有前一个节点。

只要将头结点向后移动一位就可以，这样就从链表中移除了一个头结点。

代码：

移除非头节点时，需要建立临时指针遍历链表，头指针一直是第一个结点。

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        // 删除头结点
        while (head != NULL && head->val == val) { // 注意这里不是if
            ListNode* tmp = head;
            head = head->next;
            delete tmp;
        }

        // 删除非头结点
        ListNode* cur = head;
        while (cur != NULL && cur->next!= NULL) { //避免操作空指针
            if (cur->next->val == val) {
                ListNode* tmp = cur->next;
                cur->next = cur->next->next;
                delete tmp;
            } else {
                cur = cur->next;
            }
        }
        return head;
    }
};

二、设置一个虚拟头节点再进行删除操作

该种方法旨在以一种统一的逻辑来移除链表的节点，不需要再区分头节点或非头节点。

删除当前节点需要使用到前继结点。

注意到操作完成后链表的头节点可能不再是原链表的头节点（head），故返回值为虚拟头结点的后继。

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* dummyHead = new ListNode(0); // 设置一个虚拟头结点
        dummyHead->next = head; // 将虚拟头结点指向head，这样方便后面做删除操作
        ListNode* cur = dummyHead;
        while (cur->next != NULL) {
            if(cur->next->val == val) {
                ListNode* tmp = cur->next;
                cur->next = cur->next->next;
                delete tmp;
            } else {
                cur = cur->next;
            }
        }
        head = dummyHead->next;
        delete dummyHead;
        return head;
    }
};