方法二:先找长度再找最后k
思路:
**链表不能逆向遍历,也不能直接访问。**但是对于倒数第kkk个位置,我们只需要知道是正数多少位还是可以直接遍历得到的。
具体做法:
step 1:可以先遍历一次链表找到链表的长度。
step 2:然后比较链表长度是否比k小,如果比k小返回一个空节点。
step 3:如果链表足够长,则我们从头节点往后遍历n−k次即可找到所求。
描述
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围: 0 ≤ n ≤ 1 0 5 0 \leq n \leq 10^5 0≤n≤105, 0 ≤ a i ≤ 1 0 9 0 \leq a_i \leq 10^9 0≤ai≤109, 0 ≤ k ≤ 1 0 9 0 \leq k \leq 10^9 0≤k≤109
要求:空间复杂度 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
返回值:{}
# 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:
# write code here
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