该题属于链表的简单题,只是简单的删除结点,在链表结构实现_ZZZWWWFFF_的博客-CSDN博客这篇文章里面我已经作了详解,具体操作如下
主要我们需要注意一些特殊情况的讨论
比如头结点元素就是我们要删除的、以及全部元素都要删除的情况
具体代码实现如下:
/**
* 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&&head->val==val)//头结点就要删除的情况
{
head=head->next;
}
if(!head)//头结点为空
{
return 0;
}
ListNode *pguard=head;
for(ListNode *p=pguard->next;p;)
{
if(p->val==val)//如果是,删除该元素且更新p
{
pguard->next=p->next;
p=pguard->next;
}
else//不然p和pguard都要更新
{
pguard=pguard->next;
p=p->next;
}
}
return head;
}
};
下面是我第一次写的代码,大家可以与上面的作对比(找错误)
卡在了 1 2 2 1 2 这个样例上
/**
* 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&&head->val==val)
{
head=head->next;
}
if(!head)
{
return 0;
}
ListNode *pguard=head;
for(ListNode *p=head->next;p;pguard=p,p=p->next)
{
if(p->val==val)
{
pguard->next=p->next;
}
}
return head;
}
};
相信眼尖的小伙伴已经发现了,循环条件的不同,我们在上面这个错误代码里,我们不管找没找到,我们都会同时更新pguard与p指针,这样会导致直接跳过上面样例的第二个2!!!!!!!(不信?自己调试试试)
下面是另外一种写法(与上面思路一致)
class Solution {
public ListNode removeElements(ListNode head, int val) {
//删除值相同的头结点后,可能新的头结点也值相等,用循环解决
while(head!=null&&head.val==val){//单独讨论删除头结点的情况
head=head.next;
}
if(head==null)
return head;
ListNode prev=head;
//确保当前结点后还有结点
while(prev.next!=null){
if(prev.next.val==val){
prev.next=prev.next.next;
}else{
prev=prev.next;
}
}
return head;
}
}
我们也可设置一个虚拟结点(指向头结点),便可不必单独讨论
class Solution {
public ListNode removeElements(ListNode head, int val) {
//创建一个虚拟头结点
ListNode dummyNode=new ListNode(val-1);
dummyNode.next=head;
ListNode prev=dummyNode;
//确保当前结点后还有结点
while(prev.next!=null){
if(prev.next.val==val){
prev.next=prev.next.next;
}else{
prev=prev.next;
}
}
return dummyNode.next;
}
}