4.2 移除链表元素

题目

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

示例 1：

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

提示：

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

来源：力扣（LeetCode）
链接: link.

思路

1. 创建头节点 pHead,并连接到首结点 head, 定义 pNew
2. while ( pNew到尾节点截止 ) :
   if ( pNew 的下一个节点的数据与 val 判等 // pNew 指向目标节点的前节点 ) :
   前节点 pNew 的指针域 = 目标节点的指针域 ;
   删除目标节点(需要定义 tmp 来指代删除) ;
   else : pNew = pNew->next ;
   说明 : pNew = pNew->next ;不能直接跟删节点后面,因为要删节点可能为尾节点
3. 将 head 指向 pHead 的下一个节点 ;
   删除 pHead ;
   返值

具体代码实现(C++)

/**
 * 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) {
        //1
        ListNode* pHead = new ListNode(0); //创建头节点
        pHead->next = head; 
        ListNode* pNew = pHead; // pNew 作为遍历指针
        //2
        while (pNew->next != NULL) { // pNew到尾节点截止
            if (pNew->next->val == val) { // pNew 的下一个节点的数据与 val 判等
                ListNode* tmp = pNew->next; 
                pNew->next = pNew->next->next; 
                delete tmp; 
            } else {
                pNew = pNew->next;
            }
        }
        //3
        head = pHead->next;
        delete pHead;
        return head;
    }
};