【剑指offer】JZ22 链表中倒数最后k个结点

1 问题

输入一个长度为 n 的链表，设链表中的元素的值为 ai ，返回该链表中倒数第k个节点。
如果该链表长度小于k，请返回一个长度为 0 的链表。

数据范围：$0\le n\le 10^5,0\le a_i\le 10^9,0\le k\le 10^9$

要求：空间复杂度 O(n)，时间复杂度 O(n)
进阶：空间复杂度 O(1)，时间复杂度 O(n)

例如输入{1,2,3,4,5},2时，对应的链表结构如下图所示：

其中蓝色部分为该链表的最后2个结点，所以返回倒数第2个结点（也即结点值为4的结点）即可，系统会打印后面所有的节点来比较。

示例1
输入：{1,2,3,4,5},2
返回值：{4,5}
说明：返回倒数第2个节点4，系统会打印后面所有的节点来比较。

示例2
输入：{2},8
返回值：{}

2 答案

自己写的

class Solution:
    def FindKthToTail(self , pHead: ListNode, k: int) -> ListNode:
        n = 0
        cur = pHead
        while cur:
            cur = cur.next
            n += 1
        m = n - k
        if m < 0:
            return cur  # cur 就是空链表，也可以是None，或者直接写return
        while m:
            pHead = pHead.next
            m -= 1
        return pHead

官方解

1. 快慢双指针

我们无法逆序遍历链表，就很难得到链表的倒数第k个元素，那我们可以试试反过来考虑，如果当前我们处于倒数第
k的位置上，即距离链表尾的距离是k，那我们假设双指针指向这两个位置，二者同步向前移动，当前面个指针到了链表头的时候，两个指针之间的距离还是k。虽然我们没有办法让指针逆向移动，但是我们刚刚这个思路却可以正向实施。

• step 1：准备一个快指针，从链表头开始，在链表上先走k步。
• step 2：准备慢指针指向原始链表头，代表当前元素，则慢指针与快指针之间的距离一直都是k。
• step 3：快慢指针同步移动，当快指针到达链表尾部的时候，慢指针正好到了倒数k个元素的位置。

class Solution:
    def FindKthToTail(self , pHead , k ):
        fast = pHead
        slow = pHead
        #快指针先行k步
        for _ in range(0,k): 
            if fast != None:
                fast = fast.next
            #达不到k步说明链表过短，没有倒数k
            else: 
                return None
        #快慢指针同步，快指针先到底，慢指针指向倒数第k个
        while fast:
            fast = fast.next
            slow = slow.next
        return slow

2. 先链表长度再找最后k，思路相同

链表不能逆向遍历，也不能直接访问。但是对于倒数第k个位置，我们只需要知道是正数多少位还是可以直接遍历得到的。

• step 1：可以先遍历一次链表找到链表的长度。
• step 2：然后比较链表长度是否比k小，如果比k小返回一个空节点。
• step 3：如果链表足够长，则我们从头节点往后遍历n−k次即可找到所求。

class Solution:
    def FindKthToTail(self , pHead , k ):
        n = 0
        p = pHead
        #统计链表长度
        while p: 
            n = n + 1
            p = p.next
        #长度过小，返回空链表
        if n < k: 
            return None
        p = pHead
        #遍历n-k次
        for i in range(n-k): 
            p = p.next
        return p

https://www.nowcoder.com/share/jump/9318638301698487863748