题目描述
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
解题思想
使用虚拟头节点
代码
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) {
ListNode* dummy = new ListNode(); //设置一个虚拟头节点
dummy->next = head; //将虚拟头节点指向head,这样方便后面做删除操作
ListNode* p = dummy;
while (p->next) {
if (p->next->val == val) {
ListNode* tmp = p->next;
p->next = p->next->next;
delete tmp;
}
else
p = p->next;
}
head = dummy->next;
delete dummy;
return head;
}
};