您现在的位置是:首页 >其他 >【刷题之路】LeetCode 203. 移除链表元素网站首页其他

【刷题之路】LeetCode 203. 移除链表元素

林先生-1 2023-06-16 16:00:02
简介【刷题之路】LeetCode 203. 移除链表元素

一、题目描述

原题连接: 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

二、解题

1、方法1——在原链表上动刀子

1.1、思路分析

首先我们应该能想到的就是使用一个cur指针来遍历链表,当cur-val == val时就删除cur:
在这里插入图片描述

但是这是单链表,也就意味着如果我们直接删除掉cur的话,就找不到cur后面的节点了,所以正确的做法应该是在删除之前先使用一个pre指针保存好cur的前一个节点,删除之前先让pre的next指向cur的next,然后再删除cur:
在这里插入图片描述
然后再使cur = pre->next即可。
不过有一个特殊情况就是当第一个节点的val刚好等于待删除的val时,例如:
在这里插入图片描述
这时候的cur就没有前驱节点pre了,所以这时候就应该直接执行头删。
而当我们要删除的节点刚好是链表的最后一个节点的时候,这种情况其实并不用特殊处理,因为当我们向上面一样执行完删除操作时,cur就已经为NULL了pre就已经是最后一个节点了:
在这里插入图片描述

1.2、代码实现

有了以上思路,那我们写起代码来也就水到渠成了:

struct ListNode* removeElements(struct ListNode* head, int val){
    if (NULL == head) {
        return NULL;
    }
    struct ListNode *cur = head;
    struct ListNode *pre = NULL; 
    while (cur) {
        if (val == cur->val) {
            if (cur == head) { // 头删
                head = cur->next;
                free(cur);
                cur = head;
            } else {
                pre->next = cur->next;
                free(cur);
                cur = pre->next;
            }
        } else { // 迭代地往后走
            pre = cur;
            cur = cur->next;
        }
    }
    return head;  
}

时间复杂度;O(n),n为链表的长度。
空间复杂度:O(1),我们只需要用到常数级的额外空间。

2、方法2——使用额外的链表

2.1、思路分析

我们可以创建一个新的链表,用一个新的头指针newhead指向。使用一个cur指针来遍历原链表,当cur的val不等于待删除的val时候,就将cur尾插到新链表中,当cur的val等于待删除的val时就删除cur:
在这里插入图片描述
同样的,为了保证删除节点后还能找到下一个节点,我们需要提前使用一个next指针保存cur的下一个节点:
在这里插入图片描述
而且插入新链表执行的是尾插,为了不必每次都找尾,我们需要在使用一个指针tail来保存新节点的尾节点:

在这里插入图片描述
然后每次的尾插我们就只需要执行tail->next = cur,然后执行tail = tail->next即可。
还有一个特殊情况那就只剩头插了,因为是插入第一个节点,所以此时的tail就为NULL,所以不能直接使用tail,这时候应该直接执行头插,即newhead = cur,然后再让tail = cur即可:
在这里插入图片描述

2.2、代码实现

有了以上思路,那我们写起代码来也就水到渠成了:

struct ListNode* removeElements(struct ListNode* head, int val){
    if (NULL == head) {
        return NULL;
    }
    struct ListNode *cur = head;
    struct ListNode *newhead = NULL; // 新链表的头指针
    struct ListNode *Next = NULL; // 当cur->val == val时,保存cur的下一个节点,以辅助释放cur
    struct ListNode *tail = NULL; // 保存新链表的最后一个节点
    while (cur) {
        if (cur->val == val) {
            Next = cur->next;
            free(cur);
            cur = Next;
        } else {
            if (NULL == newhead) {
                newhead = cur;
                tail = cur;
            } else {
                tail->next = cur;
                tail = tail->next;
            }
            cur = cur->next;
            tail->next = NULL;
        }
    }
    return newhead;
}
风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。