day2 移除链表元素

直接贴力扣题目

这道题代码随想录介绍了两种解法，分别是直接求解和使用虚拟头结点解题

• 第一种 直接求解，这种解法要对头结点和其他节点做区分。先贴代码再解释

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        while(head != NULL && head->val == val){
            ListNode* temp = head;
            head = head->next;
            delete temp;
        }
        ListNode* cur = head;
        while(cur->next != NULL){
            if(cur->next == val){
                ListNode* temp = cur->next;
                cur->next = cur->next->next;
                delete temp;
            }else{
                cur = cur->next;
            }
        }
        return head;
    }
};

• 首先是判断头结点是否是应删除节点，需要注意的是要用while而不是if，因为可能会有前面连续几个节点都是应删除节点的情况，删除头节点就只需要把head指向下一个就行，然后用delete删除节点内存，这里值得注意的是，整个链表应该是在堆上开辟的内存，使用delete删除非new出来的指针都不会报错。
• 然后就是非头节点的删除判断了，由于是单向链表，从前往后遍历整个链表，只需要从头结点开始判断下一个就行了，在前面头结点已经判断过了，一直往后遍历就行
• 第二种，用虚拟头结点来对节点统一判断，先贴代码

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

• 虚构头结点，就是创建一个额外的节点作为头结点，这样剩下的部分就都是除了头节点以外的部分了，就不用分开去做判断了，判断逻辑和第一种方法的非头节点一样，最后虚构头结点的下一个节点就是真实头结点了，将其赋给head，释放一下虚构头结点就结束了