力扣算法练习BM8——链表中倒数最后k个结点

题目

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

如果该链表长度小于k，请返回一个长度为 0 的链表。

数据范围：0 \leq n \leq 10^50≤n≤105，0 \leq a_i \leq 10^90≤ai​≤109，0 \leq k \leq 10^90≤k≤109

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

进阶：空间复杂度 O(1)O(1)，时间复杂度 O(n)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

复制返回值：{}

解题思路

1.遍历链表，并头插法到新链表中

2.指针移到链表最尾部，并统计节点个数

3.个数小于k，则返回空，反之取新链表前k位

题解

 

# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param pHead ListNode类 
# @param k int整型 
# @return ListNode类
#
class Solution:
    def FindKthToTail(self , pHead: ListNode, k: int) -> ListNode:
        if pHead is None: return None
        # 1.遍历链表，并头插法到新链表中
        new_pHead = pHead
        #2. 指针移到链表最尾部，并统计节点个数
        n = 1
        while new_pHead.next:
            n += 1 
            new_pHead = new_pHead.next
        #3.个数小于k，则返回空，反之取新链表前k位
        if n<k : return None
        new_pHead = pHead
        for i in range(n-k):
            new_pHead = new_pHead.next
        return new_pHead