问题
给你一个链表的头节点 head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点
示例
输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]
思路
- 如果使用C,C++编程语言的话,不要忘了还要从内存中删除这两个移除的节点,如果使用java、python的话就不需要手动管理内存了
- 那么因为单链表的特殊性,只能指向下一个节点,如果删除的是头结点又该怎么办
这里就涉及如下链表操作的两种方式:
- 直接使用原来的链表来进行删除操作
设置一个虚拟头结点在进行删除操作
- 移除头结点和移除其他节点的操作是不一样的,因为链表的其他节点都是通过前一个节点来移除当前节点,而头结点没有前一个节点。所以头结点如何移除呢,其实只要将头结点向后移动一位就可以,这样就从链表中移除了一个头结点。依然别忘将原头结点从内存中删掉。这样移除了一个头结点,在单链表中移除头结点 和 移除其他节点的操作方式是不一样,其实在写代码的时候也会发现,需要单独写一段逻辑来处理移除头结点的情况
设置一个虚拟头节点 ```cpp
直接使用原来的链表来进行移除节点操作:
class Solution { public: ListNode removeElements(ListNode head, int val) { // 删除头结点 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; } };
设置一个虚拟头结点在进行移除节点操作:
class Solution { public: ListNode removeElements(ListNode head, int val) { ListNode dummyHead = new ListNode(0); // 设置一个虚拟头结点 dummyHead->next = head; // 将虚拟头结点指向head,这样方面后面做删除操作 ListNode cur = dummyHead;
while (cur->next != NULL) {
if(cur->next->val == val) {
ListNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
} else {
cur = cur->next;
}
}
return dummyHead->next; #新的头节点
}
};
<a name="meda9"></a>
## 算法
- 初始化哨兵节点为 ListNode(0) 且设置 sentinel.next = head
- 初始化两个指针 curr 和 prev 指向当前节点和前继节点
- 当 curr != nullptr:
- 比较当前节点和要删除的节点:
- 若当前节点就是要删除的节点:则 prev.next = curr.next
- 否则设 prve = curr
- 遍历下一个元素:curr = curr.next
- 返回 sentinel.next
```java
class Solution {
public ListNode removeElements(ListNode head, int val) {
ListNode sentinel = new ListNode(0);
sentinel.next = head;
ListNode prev = sentinel, curr = head;
while (curr != null) {
if (curr.val == val) prev.next = curr.next;
else prev = curr;
curr = curr.next;
}
return sentinel.next;
}
}
- 时间复杂度:
,只遍历了一次
- 空间复杂度: