模拟(指针探测)
考虑到头节点可能被删除,所以在头节点前面加一个虚拟头节点
- 链表前面添加虚拟头节点
- 从虚拟头节点遍历,探测p->next->val是否为val,若是,则删除p->next,不是则继续探测
时间复杂度 $O(N)$,空间复杂度$O(1)$
AC代码
/**
* 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 {
public:
ListNode* removeElements(ListNode* head, int val) {
auto dummy = new ListNode(-1);
dummy->next = head;
for (auto p = dummy ; p->next ; ) {
if (p->next->val == val) {
p->next = p->next->next;
} else p = p->next;
}
return dummy->next;
}
};