【题目描述】
给你一个链表的头节点 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
输出:[]
【提示】
列表中的节点数目在范围 [0, 104] 内
1 <= Node.val <= 50
0 <= val <= 50
解题思路
-
第一种思路:先检查头结点是否是需要被删除的对象,如果是,那么将该头结点删除,并将头结点往后移动一个单位,直到遇到不需要删除的结点,确定为头结点。确定头结点后对链表中剩下的元素进行筛选。
-
第二种思路:在头结点前面生成一个虚拟头结点,这样可以用相同的方法对链表中所有的元素进行筛选,最后返回的头结点便是虚拟头结点指向的那个结点。
第一种思路
/**
* 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) {
//如果链表头部的数据域为val,则删除结点,并更新头结点为下一个结点
/*注意while循环里面的语句,容易填错
错误1:while( head->val == val)
错误原因:当链表中全部为需要删除的元素时,头结点会移动至空指针,对空指针读数据域是非法的
错误2:while( head->val == val && head != NULL)
错误原因:程序按照两个条件顺序执行,可以会遇到错误1中的情况,导致错误。
*/
while(head != NULL && head->val == val){
ListNode* temp = head;
head = head->next;
delete temp;
}
//确定了头结点之后,删除剩下需要删除的元素
ListNode* cur = head;
/*循环条件:当前结点和下一结点均不为空。*/
while(cur != NULL && cur -> next !=NULL){
//判断当前结点的下一结点是否为目标值
if(cur -> next -> val == val){
//存储即将废弃的结点
ListNode* temp = cur -> next;
//将当前结点的下一个结点指向当前结点的下两个结点
cur -> next = cur -> next -> next;
//删除即将废弃的结点
delete temp;
}else{
//移动至下一结点进行判断
cur = cur -> next;
}
}
return head;
}
};
第二种思路
/**
* 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) {
/*新增一个虚拟结点,从该节点的后面开始寻找目标值,这样可以保证整条链表的操作都是一样的*/
ListNode* VirtualNode = new ListNode(0,head);
ListNode* CurNode = VirtualNode;
while(CurNode->next != NULL){
if(CurNode->next->val == val){
ListNode* temp = CurNode->next;
CurNode->next = CurNode->next -> next;
delete temp;
}else{
CurNode = CurNode -> next;
}
}
ListNode* result = VirtualNode -> next;
delete VirtualNode;
return result;
}
};