例题
描述
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围:0 \leq n \leq 10^50≤n≤10
5
,0 \leq a_i \leq 10^90≤a
i
≤10
9
,0 \leq k \leq 10^90≤k≤10
9
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
进阶:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:
解决方案
def FindKthToTail(self , pHead, k):
if pHead is None or k <= 0:
return None
count = 0 # 判断快指针走了多少步,因为后面如果快指针走得部署小于k,则k个链表不存在
fast = slow = pHead
while fast:
fast = fast.next
if count >= k:
slow = slow.next
count += 1
if count < k:
return None
return slow
总结
链表处理我发现快慢指针用的比较多,在就是把两个链表遍历,然后把他们互相给对方玩,挺有意思得