给你一个链表的头节点 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 输出:[]
思路:本人使用的虚拟节点来移除链表元素,也可以直接针对题目给的原链表的头结点进行操作。
链表移除元素就是将指针域的关系梳理清除。代码如下:
/**
* 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) {
while(head != NULL &&head->val == val)//删除头结点
{
ListNode* q = head;
head = head->next;
delete q;
}
ListNode* p = new ListNode;//创建一个虚拟结点
p->val = NULL;//初始化虚拟结点
p->next = head;
ListNode* start = p;
while (p&&p->next )//删除非头结点
{
if (p->next->val == val)
{
ListNode* q = p->next;//删除满足条件的结点
p->next = p->next->next;
delete q;
}
else
{p = p->next;}
}
return start->next;//返回虚拟节点的next即该题中的头结点
}
};