🌟 前言
Wassup guys!我是Edison 😎
今天是 LeetCode 上的 leetcode 203. 移除链表元素
Let’s get it!
1. 题目分析
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足
Node.val == val
的节点,并返回 新的头节点 。
示例 1:
示例 2:
示例 3:
2. 题目图解
这道题很简单,还是采用 双指针 法,但是要考虑几种情况
(1)常规情况
(2)链表中有连续的 val
(3)头节点就是 val
🍑 常规情况
定义一个 prev 指针指向 NULL;再定义一个 cur 指针指向头节点
第 1 步:cur 指向的元素不等于 6,那么 prev 和 cur 依次向后挪动(如图所示👇)
第 2 步:cur 指向的元素不等于 6,那么 prev 和 cur 依次向后挪动(如图所示👇)
第 3 步:此时 cur 指向的元素等于 6,那么元素 6 置为 NULL,然后让 prev 的 next 指向元素 3,cur 的 next 也指向元素 3 (如图所示👇)
第 4 步:cur 指向的元素不等于 6,那么 prev 和 cur 依次向后挪动(如图所示👇)
第 5 步:cur 指向的元素不等于 6,那么 prev 和 cur 依次向后挪动(如图所示👇)
第 6 步:此时 cur 指向的元素等于 6,那么元素 6 置为 NULL,然后让 prev 的 next 指向 cur 的 next(如图所示👇)
第 7 步:当 cur 指向的元素为 空 时,循环就终止了(如图所示👇)
✨动图演示
🍑 连续 val 情况
还是定义一个 prev 指针指向 NULL;再定义一个 cur 指针指向头节点(如图所示👇)。
第 1 步:cur 指向的元素不等于 6,那么 prev 和 cur 依次向后挪动(如图所示👇)
第 2 步:cur 指向的元素不等于 6,那么 prev 和 cur 依次向后挪动(如图所示👇)
第 3 步:此时 cur 指向的元素等于 6,那么删除该元素,然后让 prev 的 next 指向 cur 的 next,cur 指向下一个元素(如图所示👇)
第 4 步:此时 cur 指向的元素还是等于 6,那么删除该元素,然后让 prev 的 next 指向 cur 的 next,cur 指向下一个元素(如图所示👇)
第 5 步:此时 cur 指向的元素还是等于 6,那么删除该元素,然后让 prev 的 next 指向 cur 的 next,cur 指向下一个元素(如图所示👇)
第 6 步:此时 cur 指向的元素不是 6,那么 prev 和 cur 依次向后挪动(如图所示👇)
第 7 步:此时 cur 指向的元素等于 6,那么删除该元素然后让 prev 的 next 指向 cur 的 next,cur 指向 空,终止循环(如图所示👇)
可以看到,如果链表中有连续的 val,那么做法也是和常规情况一样的。
🍑 头节点就为 val
还是定义一个 prev 指针指向 NULL;再定义一个 cur 指针指向头节点(如图所示👇)。
此时可以发现一个问题,如果是按照常规情况来算的是,cur 指向的元素等于 6,那么删除该元素,然后让 prev 的 next 指向 cur 的 next,cur 指向下一个元素。
但是此时 prev 指向的元素为 空, cur 的 next 怎么能够赋值给 空 呢?显然是不行的,得另想办法!
其实很简单,我们再定义一个 curHead 指针指向头节点(如图所示👇)。
这里直接用一个动图来演示整个过程👇
3. 代码实现
📝 接口代码
struct ListNode* removeElements(struct ListNode* head, int val){
struct ListNode* prev, *cur;
prev = NULL;
cur = head;
// 循环的结束条件就是cur等于空
while (cur) {
// cur不等于val
if (cur->val != val) {
prev = cur; //直接把cur的地址赋给prev,相当于把prev移动到cur的位置上
cur = cur->next; // cur也向后移动一个位置
}
else { // 当cur等于val
struct ListNode* curHead = cur->next;
// 为什么 curHead 的内容为 cur->next, 而不是cur呢?
// 假设头节点就等于val,那么删除val以后, 就相当于头删
// 头删之后,如果curHead是内容是cur的地址的话,那么久找不到后面的链表了
// 所以curHead存的应该是cur->next, 也就是第二个节点的地址
if (prev == NULL) { // 头节点为val为val的情况
free(cur); // 释放掉cur
head = curHead; // 更新头节点
cur = curHead; // 然后把新的头节点的地址赋给cur
}
else { // 头节点不为val为val的情况
free(cur); // 释放val元素
prev->next = curHead; // 因为curHead存的是cur的next,也就是cur的下一个节点的地址,所以直接赋给prev的next
cur = curHead; // 然后让cur指向下一个节点的地址
}
}
}
return head;
}
🌟 提交结果