给你一个链表的头节点 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 输出:[]
实现代码:
直接使用原来的链表来进行移除节点的操作
class Solution {
public:
ListNode* removeElements(ListNode* head, int val)
{
//删除头节点
while(head!=NULL&&head->val==val)
{
ListNode*temp=head;
head=head->next;
delete temp;
}
ListNode* p=head;
//删除非头结点
while(p!=NULL&&p->next!=NULL)
{
if(p->next->val==val)
{
ListNode* temp=p->next;
p->next=p->next->next;
delete temp;
}
else
{
p=p->next;
}
}
return head;
}
};
由于删除结点需要知道此结点的前驱结点,所以在删除过程中,删除头节点的代码就与其他结点不同,因为头结点没有前驱结点,要单独判断所要删除的结点是否为头节点,在删除头结点时要注意用的是while而不是if,因为在删除头结点后,新的头结点可能也为要删除的目标结点。同时要注意删除结点时指针变化的顺序,防止部分链表的丢失。注意这里头节点也储存元素。
设置一个虚拟头结点在进行移除节点的操作
class Solution {
public:
ListNode* removeElements(ListNode* head, int val)
{
ListNode* newhead=new ListNode();//建立一个新头节点
newhead->next=head;
ListNode* p=newhead;//用于遍历的指针
while(p->next!=NULL)
{
if(p->next->val==val)
{
ListNode* temp=p->next;
p->next=p->next->next;
delete temp;
}
else
{
p=p->next;
}
}
head=newhead->next;
delete newhead;
return head;
}
};
创建新的空头结点,可以让删除结点代码变得统一起来,返回的时候记得返回真正的头节点,而不是我们新创建的结点即可。