leetcode刷题

leetcode刷题 ,第1张

1. 移除链表元素

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

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/872778.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-13
下一篇 2022-05-13

发表评论

登录后才能评论

评论列表(0条)

保存