力扣日记:【链表篇】移除链表元素
日期:2023.8.8
参考:代码随想录、力扣
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, 10^4] 内
- 1 <= Node.val <= 50
- 0 <= val <= 50
题解
/**
* 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 {
#define SOLUTION 2
public:
ListNode* removeElements(ListNode* head, int val) {
#if SOLUTION == 1 //
ListNode* ptr = head;
while (ptr != nullptr) {
// 开头
if (head->val == val) {
ListNode* tmp = head;
head = head->next;
ptr = head;
delete tmp;
}
else if (ptr->next != nullptr && ptr->next->val == val) {
ListNode* tmp = ptr->next;
ptr->next = (ptr->next)->next;
delete tmp;
} else {
ptr = ptr->next; // 没有删除元素,才往后移
}
}
return head;
#elif SOLUTION == 2 // 直接使用原来的链表来进行移除节点操作(代码随想录ver):
// 头节点与非头节点单独操作(而不会像SOLUTION 1 混在一起处理)
// 先删除头结点
while (head != NULL && head->val == val) { // 注意这里不是if
ListNode* tmp = head;
head = head->next;
delete tmp;
}
// 删除非头结点
ListNode* cur = head;
while (cur != NULL && cur->next!= NULL) {
if (cur->next->val == val) {
ListNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
} else {
cur = cur->next;
}
}
return head;
#elif SOLUTION == 3 // 使用虚拟头节点(代码随想录ver)
// 设置虚拟头节点,指向头节点
ListNode* dummyHead = new ListNode(0, head);
// 则对头节点和非头节点的处理一致
ListNode* cur = dummyHead;
while (cur != NULL && cur->next != NULL) {
if (cur->next->val == val) {
ListNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
} else {
cur = cur->next;
}
}
head = dummyHead->next;
delete dummyHead; // 记得删除创建的虚拟头节点
return head;
#endif
}
};
执行结果
状态:通过
时间复杂度:O(n)
空间复杂度:O(1)
思路总结
- 链表理论基础
- 移除链表元素分为两种情况,头节点和非头节点。
- 对于头节点:将头节点指向其下一个节点即可;
- 对于非头节点:依赖于其前面的节点(所以要提前判断,即
cur->next->val == val
)进行删除,即cur->next = cur->next->next
。 - 对于非头节点,只有在没有删除元素时,当前指针才向后移。
- 如果要使得头节点与非头节点的处理一致,则设置虚拟头节点指向原先的头节点。
- 记得在移除链表元素时要手动清除内存(设置的虚拟头节点在程序结束时也要清除内存)