203.移除链表元素
给你一个链表的头节点 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
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-linked-list-elements
无虚拟头结点
思路:将指针指向头结点,如果指针的下一个元素等于val,则将指针指向下下个元素,并删除与val相等的节点。如果不等于val,则至臻后移。
但是,每次删除的都是指针的下一位元素,所有如果头结点也需要删除就需要对头结点进行处理。
/**
* 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!=nullptr&&head->val==val)//删除头结点
{
ListNode *q=head;
head=head->next;
delete q;
}
ListNode *p=head;
while(p!=nullptr&&(p->next)!=nullptr)//删除中间结点
{
if(p->next->val==val)
{
ListNode *q=p->next;
p->next=q->next;
delete q;
}
else
{
p=p->next;
}
}
return head;
}
};
有虚拟头结点
思路:在头结点之前设计一个虚拟头结点,使指针从虚拟头结点之后的位置开始删除,则不会出现头结点无法删除。
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode *dummynode=new ListNode;
dummynode->next=head;
ListNode *p=dummynode;
while(p!=nullptr&& p->next!=nullptr)
{
if(p->next->val==val)
{
ListNode *q=p->next;
p->next=q->next;
delete q;
}
else
{
p=p->next;
}
}
return dummynode->next;
}
};