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

这样就可以解决了。

发布了29 篇原创文章 · 获赞 28 · 访问量 295

猜你喜欢

转载自blog.csdn.net/weixin_45398265/article/details/104807681
今日推荐