从列表末尾删除第 N 个节点 LeetCode

Question

我正在参加Remove Nth Node from End of List LeetCode 挑战：

给定链表的head，从链表的末尾删除n^th节点并返回它的头。

跟进：你能一次性完成吗？

我的代码在某些测试用例上失败了：

[1,2]
2

我得到了这份报告：

Output [1]
Expected [2]

我已经阅读了解决方案，我的解决方案似乎与所写的相同，但我不知道我的错误在哪里。我的代码如下：

def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
    pointer1 = head
    pointer2 = head
    count = 0
    if not pointer1.next:
        return pointer1.next
    while pointer2 is not None:
        if count > n:
            pointer1 = pointer1.next
        pointer2 = pointer2.next
        count += 1
    pointer1.next = pointer1.next.next
    return head

我已经在这个问题上待了半个小时，我不知道如何解决它。有人可以帮忙吗？

Answer 1

当n 等于列表中的节点数时，您的解决方案将失败。在这种情况下，您应该从列表中删除第一个节点，因此您应该返回 head.next。

注意：if 在 while 循环测试之前的 if 是针对这种情况的一个非常具体的实例，即当列表只有一个元素时。但如上所述，此操作应适用于比if 中确定的更通用的情况。

因此，可以删除以下if 条件：

if not pointer1.next:
    return pointer1.next

并在while 循环之后添加一个条件：

def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
    pointer1 = head
    pointer2 = head
    count = 0
    while pointer2 is not None:
        if count > n:
            pointer1 = pointer1.next
        pointer2 = pointer2.next
        count += 1
    if count > n:
        pointer1.next = pointer1.next.next
        return head  
    else:
        return head.next