LeetCode:双指针3(Python3)[19, 876]

19. 删除链表的倒数第 N 个结点给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。

进阶：你能尝试使用一趟扫描实现吗？
示例1

输入：head = [1,2,3,4,5], n = 2
输出：[1,2,3,5]

思路
链表的特点就是只能从前往后去遍历,那么要找到倒数第n个节点所在的位置,可以设置双指针,让一个指针(fast)先走n步,然后两个指针再一起往后走,等到fast指针走到最后一个节点的时候,slow指针所在的位置就是倒数第n个节点的前一个节点了.
注意,这里如果需要删除的是第一个节点,那么其并没有前向节点,需要单独考虑,很简单,就是如果fast走到最后一个节点slow还没动,那么删除的就是头节点了

正确代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeNthFromEnd(self, head: ListNode, n: int) -> ListNode:
        slow, fast = head, head
        for i in range(n):
            fast = fast.next

        if fast is None:
            return head.next
        while fast.next:
            fast = fast.next
            slow = slow.next

        slow.next = slow.next.next
        return head

876. 链表的中间结点给定一个头结点为 head 的非空单链表，返回链表的中间结点。

如果有两个中间结点，则返回第二个中间结点。

示例 1

输入：[1,2,3,4,5]
输出：此列表中的结点 3 (序列化形式：[3,4,5])
返回的结点值为 3 。 (测评系统对该结点序列化表述是 [3,4,5])。
注意，我们返回了一个 ListNode 类型的对象 ans，这样：
ans.val = 3, ans.next.val = 4, ans.next.next.val = 5, 以及 ans.next.next.next = NULL.

示例2

输入：[1,2,3,4,5,6]
输出：此列表中的结点 4 (序列化形式：[4,5,6])
由于该列表有两个中间结点，值分别为 3 和 4，我们返回第二个结点。

思路
设置双指针,快指针是慢指针的两倍移动速度,如果链表元素数量为偶数,那么快指针走到尾节点后一个时候,慢指针指向中间节点.如果链表元素为奇数,那么快指针走到尾节点时,慢指针指向中间节点

正确代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def middleNode(self, head: ListNode) -> ListNode:
        fast = head
        while fast.next is not None and fast.next.next is not None:
            fast = fast.next.next
            head = head.next
        if fast.next:
            return head.next
        else:
            return head