给你一个链表的头节点?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
法一:添加头节点
这个题没有头结点的(也可以收他头节点存储数据了),所以如果不单独考虑头结点,就比如val为7,链表为7 7 7 这个情况,直接去做的话,就会处理不了第一个结点,最后输出肯定是单独一个7.所以我们可以new一个虚拟节点当作头结点。
/**
* 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 *vr = new ListNode(val-1);
vr->next = head;
ListNode *pre = vr,*p = vr->next;
while(p){
if(p->val == val){
ListNode *tmp = p;
pre -> next = p->next;
p = p->next;
delete tmp;
}else{
p = p->next;
pre = pre->next;
}
}
return vr->next;
}
};
法二:单独处理头结点
第一步先判断头结点是不是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&&head->val==val){
ListNode *tmp = head;
head = head -> next;
delete tmp;
}
ListNode *p = head;
while(p&&p->next){
if(p->next->val == val){
ListNode *tmp = p->next;
p->next = tmp->next;
delete tmp;
}else{
p = p->next;
}
}
return head;
}
};