Leetcode:203. 移除链表元素(C++)

给你一个链表的头节点 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;

    }
};

创建新的空头结点,可以让删除结点代码变得统一起来,返回的时候记得返回真正的头节点,而不是我们新创建的结点即可。

经验分享 程序员 微信小程序 职场和发展