leetcode 删除链表倒数第n个节点(一次扫描)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/weixin_36372879/article/details/86242619

题目描述
给定一个链表,删除链表的倒数第 n 个节点,并且返回链表的头结点。
在这里插入图片描述


算法思想

采用一次扫描的办法,需要定位到删除节点的前面的节点
因此指针p往后挪动的个数为n+1次
注意如果p挪动之后,碰到空,说明删除的是头结点
使用两个指针,一个从头还是挪动,一个从p开始挪动,知道p为空,就找到了前面的指针,之后直接删除即可


通过代码

class Solution:
    def removeNthFromEnd(self, head, n):
        """
        :type head: ListNode
        :type n: int
        :rtype: ListNode
        """
        if head == None or n < 1:
            return None
        p = head
        for i in range(n):
            p = p.next
        #如果p为None,那么需要删除头结点
        if p == None:
            tempHead = head
            head = head.next
            del tempHead
            return head
        preNode = head
        while p.next:
            preNode = preNode.next
            p = p.next
        deleteNode = preNode.next
        preNode.next = deleteNode.next
        del deleteNode
        return head

运行结果
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_36372879/article/details/86242619