移除链表元素(双指针/虚拟头指针)

该题属于链表的简单题,只是简单的删除结点,在链表结构实现_ZZZWWWFFF_的博客-CSDN博客这篇文章里面我已经作了详解,具体操作如下

主要我们需要注意一些特殊情况的讨论

比如头结点元素就是我们要删除的、以及全部元素都要删除的情况

具体代码实现如下:

/**
 * 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) {
        while(head&&head->val==val)//头结点就要删除的情况
        {
            head=head->next;
        }
        if(!head)//头结点为空
        {
            return 0;
        }        
        ListNode *pguard=head;
        for(ListNode *p=pguard->next;p;)
        {
            if(p->val==val)//如果是,删除该元素且更新p
            {
                pguard->next=p->next;
                p=pguard->next;
            }
            else//不然p和pguard都要更新
            {
                pguard=pguard->next;
                p=p->next;
            }
        }
        return head;
    }
};

下面是我第一次写的代码,大家可以与上面的作对比(找错误) 

卡在了 1 2 2 1     2       这个样例上

/**
 * 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) {
        while(head&&head->val==val)
        {
            head=head->next;
        }
        if(!head)
        {
            return 0;
        }        
        ListNode *pguard=head;
        for(ListNode *p=head->next;p;pguard=p,p=p->next)
        {
            if(p->val==val)
            {
                pguard->next=p->next;
            }
        }
        return head;
    }
};

相信眼尖的小伙伴已经发现了,循环条件的不同,我们在上面这个错误代码里,我们不管找没找到,我们都会同时更新pguard与p指针,这样会导致直接跳过上面样例的第二个2!!!!!!!(不信?自己调试试试)

下面是另外一种写法(与上面思路一致)

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        //删除值相同的头结点后,可能新的头结点也值相等,用循环解决
        while(head!=null&&head.val==val){//单独讨论删除头结点的情况
            head=head.next;
        }
        if(head==null)
            return head;
        ListNode prev=head;
        //确保当前结点后还有结点
        while(prev.next!=null){
            if(prev.next.val==val){
                prev.next=prev.next.next;
            }else{
                prev=prev.next;
            }
        }
        return head;
    }
}

我们也可设置一个虚拟结点(指向头结点),便可不必单独讨论

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        //创建一个虚拟头结点
        ListNode dummyNode=new ListNode(val-1);
        dummyNode.next=head;
        ListNode prev=dummyNode;
        //确保当前结点后还有结点
        while(prev.next!=null){
            if(prev.next.val==val){
                prev.next=prev.next.next;
            }else{
                prev=prev.next;
            }
        }
        return dummyNode.next;
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ZZZWWWFFF_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值