C语言每日一题(26)移除链表元素

2024-01-23 15:06:04 浏览数 (2)

力扣网 203. 移除链表元素

题目描述

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

思路分析

针对如图的普通情况,不能简单的遍历到对应位置然后进行释放,一方面会丢掉下一个结点的地址,同时上一个结点的成员指针也会变成野指针。基本的方法是,每次循环,都保留该节点的下一个结点地址,也要保留上一个结点的地址(该指针最开始指向NULL),如果该节点的值是需要删除值,让前一个结点的指针指向该节点的下一个结点后再进行释放,如果不是就把当前结点保留,继续遍历下去,

另外一种特殊情况是当头结点就是需要删除的值,如果直接让前一个结点的指针指向该节点的下一个结点,那就是对空指针的非法引用了,是不允许的,所以这里还要再加一个判断条件,即当前一个结点(设为pre)不为空时,让前一个结点的指针指向该节点的下一个结点,否则,直接将下一个结点赋给头结点,因为释放完后头结点也为野指针,此时需要进行赋值。

代码语言:javascript复制
struct ListNode* removeElements(struct ListNode* head, int val) {
    struct ListNode* pre=NULL;//前驱指针,最开始为NULL
    struct ListNode* cur=head;//对应指针
    while(cur)
    {
        if(cur->val==val)
        {
            struct ListNode* next=cur->next;//用来保存下一个结点
            free(cur);
            if(pre)//如果pre不为空,直接让pre-》next指向next即可            {
                pre->next=next;
            }
            else
            {
                head=next;//为空则将头结点指向next
            }
            cur=next;
        }
        else//不是对应值,将当前指针赋给pre成为下一个指针的前驱指针,cur往下遍历
        {
            pre=cur;
            cur=cur->next;
        }
    }
    return head;//返回头结点
    
}

0 人点赞