【LeetCode 203.移除链表元素】——链表，虚拟头结点法

最新推荐文章于 2023-03-14 15:22:32 发布

木瓜星灵TT

最新推荐文章于 2023-03-14 15:22:32 发布

阅读量299

点赞数 4

分类专栏： LeetCode学习笔记文章标签：链表 leetcode 数据结构算法 c++

本文链接：https://blog.csdn.net/weixin_51630192/article/details/126473705

版权

LeetCode学习笔记专栏收录该内容

37 篇文章 0 订阅

订阅专栏

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

题目链接

思考：

本题是一道经典的对链表这种数据结构的考察，主要考察了对链表中元素删除的操作。删除某个节点，其实只需要将上个节点指向该结点的下个结点即可。在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) {
       //delete head;
       while (head != NULL && head->val==val)
       {
            ListNode* tmp = head;
            head = head->next;
            delete tmp;
       }
       //delete other nodes;
       ListNode* cur=head;//将cur指向头指针开始遍历
       while(cur!=NULL && cur->next!=NULL)
       {
        if (cur->next->val == val) { //实际操作删除修改的是cur->next即cur后一个元素
                ListNode* tmp = cur->next;
                cur->next = cur->next->next;
                delete tmp;
            } 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* newHead = new ListNode(0);//设置虚拟头结点
        newHead->next = head;//原来的头结点称为第二位
        ListNode* cur = newHead;//剩下的操作只需考虑一种删除
        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 = newHead->next;//还原头结点
        delete newHead;
        return head;
    }
};