leetcode删除链表的倒数第N个节点

给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
示例:
给定一个链表: 1->2->3->4->5, 和 n = 2.
当删除了倒数第二个节点后,链表变为 1->2->3->5.
说明:
给定的 n 保证是有效的。
进阶:
你能尝试使用一趟扫描实现吗?
最先想到的做法一定是先找出链表的长度,然后正向找到倒数第N个节点。这样思路简单,但是效率估计就没有那么高效了。
先展示一下我用第一种做法写出的代码:

class Solution(object):
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        p=q=head
        s=0
        if p==None :
            return head
        while p:
            s+=1
            p=p.next
        s=s-n
        if s==0:
            head=head.next
            return head
        while s>1:
            s-=1
            q=q.next
        q.next=q.next.next
        return head

更改一下思路,设立两个快慢指针,快指针先走n步,之后设立循环,在快指针指到空节点之前,快慢指针一起走,这样慢指针在循环完之后会留下n个节点,也就得到了倒数第N个节点。可是,这里需要注意一个问题,如果,直接快慢指针走下去的话,会出现越界的问题。比如,链表长度为2,要删除倒数第N个节点,按照上面的思路,走n步后,快指针应该指在最后一个节点,那在循环时,我们的判断语句肯定是:
while l1.next:
看代码:

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        l1=l2=head
        if l1==None:
            return head
        if l1.next==None and n==1:
            return None
        for i in range(n):
            l1=l1.next
        while l1.next:
            l1=l1.next
            l2=l2.next
        l2.next=l2.next.next
        return head

这样l1本来就在最后一个节点,哪里还有next。

所以有一个解决办法,添加一个头节点:

class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        head2=ListNode(0)
        head2.next=head
        l1=l2=head2
        if l1.next==None or (l1.next==None and n==1):
            return None
        for i in range(n):
            l1=l1.next
        while l1.next:
            l1=l1.next
            l2=l2.next
        l2.next=l2.next.next
        return head2.next

这样就可以解决了。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值