《代码随想录》(6)移除链表元素

LeeCode题号： 203

---

【题目描述】

给你一个链表的头节点 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
输出：[]

【提示】

列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50

---

解题思路

• 第一种思路：先检查头结点是否是需要被删除的对象，如果是，那么将该头结点删除，并将头结点往后移动一个单位，直到遇到不需要删除的结点，确定为头结点。确定头结点后对链表中剩下的元素进行筛选。
  
  
  

• 第二种思路：在头结点前面生成一个虚拟头结点，这样可以用相同的方法对链表中所有的元素进行筛选，最后返回的头结点便是虚拟头结点指向的那个结点。
  

---

第一种思路

/**
 * 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) {
        //如果链表头部的数据域为val，则删除结点，并更新头结点为下一个结点
        /*注意while循环里面的语句，容易填错
        错误1：while( head->val == val) 
        错误原因：当链表中全部为需要删除的元素时，头结点会移动至空指针，对空指针读数据域是非法的
        错误2：while( head->val == val && head != NULL) 
        错误原因：程序按照两个条件顺序执行，可以会遇到错误1中的情况，导致错误。
        */
        while(head != NULL && head->val == val){
            ListNode* temp = head;
            head = head->next;
            delete temp;
        }

        //确定了头结点之后，删除剩下需要删除的元素
        ListNode* cur = head;
        /*循环条件：当前结点和下一结点均不为空。*/
        while(cur != NULL && cur ->  next !=NULL){
            //判断当前结点的下一结点是否为目标值
            if(cur -> next -> val == val){
                //存储即将废弃的结点
                ListNode* temp = cur -> next;
                //将当前结点的下一个结点指向当前结点的下两个结点
                cur -> next = cur -> next -> next;
                //删除即将废弃的结点
                delete temp;
            }else{
                //移动至下一结点进行判断
                cur = cur -> next;
            }

        }

        return head;
    }
};

第二种思路

/**
 * 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* VirtualNode = new ListNode(0,head);
        ListNode* CurNode = VirtualNode;
        while(CurNode->next != NULL){
            if(CurNode->next->val == val){
                ListNode* temp = CurNode->next;
                CurNode->next = CurNode->next -> next;
                delete temp;
            }else{
                CurNode = CurNode -> next;
            }

        }

        ListNode* result = VirtualNode -> next;
        delete VirtualNode;
        return result;

    }
};

---