力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
提示:力扣中链表的 head 是头指针,而不是头结点,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) {
if (!head)
return head;
head->next=removeElements(head->next,val);
return head->val==val?head->next:head;
}
};
这段代码的功能是从一个单链表(singly-linked list)中删除所有值为 val 的节点。
首先,定义了一个单链表的节点结构 ListNode,包含一个整数值 val 和一个指向下一个节点的指针 next。
然后,定义了一个名为 Solution 的类,其中包含一个公共方法 removeElements,该方法接受一个单链表的头结点 head 和一个整数 val 作为参数。
在 removeElements 方法中,首先检查头结点是否为空,如果为空,则直接返回 nullptr。
接下来,使用递归的方式删除值为 val 的节点。递归调用 removeElements 方法,并将当前节点的下一个节点作为参数传递给递归调用。递归调用的结果将被赋值给当前节点的 next 指针,从而删除值为 val 的节点。
最后,返回当前节点。如果当前节点的值等于 val,则返回当前节点的下一个节点;否则,返回当前节点本身。
这样,通过递归的方式遍历整个单链表,并将删除的节点指针返回给上一层递归调用,最终实现删除所有值为 val 的节点的功能。
- 时间复杂度:O(1)
- 空间复杂度:O(n)
方法二 直接原链表移除删除
/**
* 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* L=new ListNode();
L->next=head;
ListNode* pre=L;
ListNode* p=head;
while(p!=nullptr){
if(p->val!=val){
pre=p;
p=p->next;
}else{
p=p->next;
pre->next=p;
}
}
return L->next;
}
};
这段代码的功能是从一个单链表中删除所有值为 val 的节点,并返回新的头结点。
首先,定义了一个单链表的节点结构 ListNode,包含一个整数值 val 和一个指向下一个节点的指针 next。
然后,定义了一个名为 Solution 的类,其中包含一个公共方法 removeElements,该方法接受一个单链表的头结点 head 和一个整数 val 作为参数。
在 removeElements 方法中,首先创建了一个新的空节点 L,并将其 next 指针指向头结点 head。然后,定义了两个指针变量 pre 和 p,分别初始化为新节点 L 和头结点 head。
接下来,使用一个循环遍历整个链表。在循环中,首先检查当前节点 p 的值是否等于 val。如果不等于 val,则将 pre 指向当前节点 p,并更新 p 为下一个节点。如果等于 val,则将 pre 的 next 指针设置为 p 的下一个节点,跳过当前节点 p。
最后,返回新节点 L 的 next 指针,即新的头结点。
- 时间复杂度:O(n)
- 空间复杂度:O(1)