Leetcode-203: 移除链表元素

题目链接

https://leetcode-cn.com/problems/remove-linked-list-elements/

题目

给你一个链表的头节点 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

思路

一般情况下，需要对头结点和非头结点分情况讨论。这里我们可以设置一个虚拟头结点，这样就可以把所有结点统一处理，等处理完了再将虚拟头结点删除即可。

C++ Code

/**
 * 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* Vir= new ListNode(0,head);
        ListNode* cur=Vir; //当前指针指向虚拟头结点
        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=Vir->next;
        delete Vir;
        return head;
    }
};