203
题目:给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]
思路:设置哑结点,这样头结点和其他结点处理一样。
笔记:
C++
链表可以直接赋空值,也可以指向空值。
// 赋空值
ListNode* head = NULL;
// 指向空
p->next = NULL;
Python:
由于Python自动管理内存,所以不需要删除结点直接断线就行,不需要delete *** 作,也就不需要设tmp。
C++:
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
// 设置虚拟结点
ListNode* dummy = new ListNode(0);
// 指向头结点
dummy->next = head;
ListNode* p = dummy;
while(p->next != NULL){
if(p->next->val == val){
ListNode* tmp = p->next;
p->next = p->next->next;
delete tmp;
}else{
p = p->next;
}
}
//不要忘记最后重新赋head,因为原head有可能已经被删了
head = dummy->next;
delete dummy;
return head;
}
};
Python:
class Solution:
def removeElements(self, head: ListNode, val: int) -> ListNode:
dummy = ListNode(next=head)
p = dummy
while(p.next != None):
if(p.next.val == val):
p.next = p.next.next
else:
p = p.next
return dummy.next
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)