每日leetcode:删除链表的倒数第n个节点(python)

最直接的想法肯定是遍历一遍求出长度,然后用这个这个长度减去n-1定位到要删除的节点,然后再次遍历用一个变量进行计数,然后删除。

这是最低效的办法,提高算法效率的一个基本思想就是:在一次遍历中完成多个任务,这需要在设计算法时抓住题目要求的本质,剔除那些看完题目直接浮现在脑海但是不必要的步骤,同时在必要的步骤中尽可能多地搭载任务。在实现算法时注意各种可能拖慢程序的实现方法。这些注意点可以在计算两数之和的每次求和计算进位等处理方法中可以看出。

很明显,这里是必须要进行遍历的,但是能不能在一次遍历中就定位到要删除的节点?考虑到遍历结束条件是Node == None,往前n个就是要删除的节点,那么可以用另一个变量记录Node前第n个节点,这样就能一次遍历实现了。

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        CurNode = head
        CurList = head
        m = 0
        while CurNode != None:
            m += 1
            CurNode = CurNode.next
            if m > n:
                preCurList = CurList
                CurList = CurList.next
        if m == n:
            return head.next
        preCurList.next = CurList.next                
        return head

这个实现方法很慢,平均76ms,只打败了27%的用户,这里拖慢程序的一个核心点就是循环中那个if m> n:只需要判断出一次m>n,明显后面的每一次循环内部都不需要再判断了,所以该怎么改?

这种在循环中的只需要一次的条件判断,可以处理成两个循环,第一个循环以条件判断为循环结束条件,第二个循环则以原循环退出条件为条件。

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        CurNode = head
        CurList = ListNode(0)
        CurList.next = head
        while n > 0:
            n -= 1
            CurNode = CurNode.next
        if CurNode == None:
            return head.next
        while CurNode != None:
            CurNode = CurNode.next
            CurList = CurList.next              
        CurList.next = CurList.next.next
        return head

这次平均60ms,大概打败56%的用户,这里除了上述的优化之外,还有一个小技巧就是pre指针的设置,这里不是记录要删除的节点,而是记录其前驱,这样很明显可以节省一个变量空间以及运行时间。

再优化就是参考别人的题解了

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        CurList = ListNode(0)
        CurList.next = head
        start = CurList
        end = CurList
        while n != 0:
            n -= 1
            start = start.next
        while start.next != None:
            start = start.next
            end = end.next              
        end.next = end.next.next
        return CurList.next

这次优化的主要方面就是:直接去掉了两个循环之间的if,设置了pre除了在提升效率之外,还能防止一些边界情况

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值