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,则至臻后移。 但是,每次删除的都是指针的下一位元素,所有如果头结点也需要删除就需要对头结点进行处理。
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;
}
};
|