leetcode日记:19. 删除链表的倒数第N个节点

GreenPill 2020-12-01 00:29:37

分类专栏： leetcode-python-top100 文章标签： leetcode 数据结构链表

最后发布:2020-12-01 00:29:37首次发布:2020-12-01 00:29:37

本文链接：https://blog.csdn.net/u013992365/article/details/110412541

版权

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

题目

给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。

示例：

给定一个链表: 1->2->3->4->5, 和 n = 2.

当删除了倒数第二个节点后，链表变为 1->2->3->5.
说明：

给定的 n 保证是有效的。

进阶：

你能尝试使用一趟扫描实现吗？

题解

一看这题面,一次遍历,我直接掏出两个指针,一个用来记录终点,一个用来找到目标节点位置.
在这里插入图片描述

另有两种情况需要额外处理下

节点数为1的情况,这种情况需要返回空节点
需要删除的节点为头结点的情况,这种情况需要返回头结点的下一个节点
贴下代码:

# Definition for singly-linked list.
class ListNode:
    def __init__(self, val=0, next=None):
        self.val = val
        self.next = next
def removeNthFromEnd(head: ListNode, n: int) -> ListNode:
    if head.next == None:
        return None

    target_node = head
    tail_node = head
    index = 0
    while tail_node.next != None:
        index += 1
        if index > n:
            target_node = target_node.next
        tail_node = tail_node.next
    if index == n:
        head = head.next
    target_node.next = target_node.next.next
    return head