【LeetCode】203.移除链表元素

思路

遍历链表节点,找出val并删除。

方法一 直接处理

直接在原链表上修改,这时候我们需要考虑头结点是否与val相等。

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        //1.处理头结点值为val的情况
        while (head != nullptr && head->val == val)
        {
            ListNode* tmp = head;
            head = head->next;
            delete tmp;
        }
        //2.若头结点为空,直接返回就好
        if (head == nullptr)
            return nullptr;
        
        //3.下面只剩下头结点不为空且头结点值不等于val的情况了
        ListNode* prev = head;
        ListNode* cur = head->next;
        while (cur != nullptr)
        {
            if (cur->val == val)
            {
                ListNode* tmp = cur;
                prev->next = cur->next;
                cur = cur->next;
                delete tmp;
            }
            else
            {
                prev = cur;
                cur = cur->next;
            }
        }
        return head;
    }
};

方法二 虚拟头结点

即在原本头结点前加一个哨兵位,这样我们就不需要再单独考虑原头结点的情况了,推荐用此方法。

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {

        if (head == nullptr)
            return nullptr;
        ListNode* dummy = new ListNode(0);
        dummy->next = head;
        
        //下面处理的是原头结点不为空的情况,相当于方法一的3
        ListNode* prev = dummy;
        ListNode* cur = head;
        while (cur != nullptr)
        {
            if (cur->val == val)
            {
                ListNode* tmp = cur;
                cur = cur->next;
                prev->next = cur;
                delete tmp;
            }
            else
            {
                prev = cur;
                cur = cur->next;
            }
        }
        return dummy->next;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值