剑指offer JZ14:链表中倒数第k个结点
问题
输入一个链表,输出该链表中倒数第k个结点
思路
首先,看到问题的第一反应就是将链表的每个结点按顺序输出到列表里,再将列表中的倒数第k项输出即可,但是需要消耗0(N)的空间。
这里运用一个非常巧妙的双指针算法:要输出倒数第k项,只需要让一个指针先跑k-1项,再让两个指针同时向后移动,当第一个指针到达队尾时,第二个指针指向的自然就是倒数第k项了。
注意对于k>n等特殊情况的讨论
代码及解释
class Solution:
def FindKthToTail(self, head, k):
# write code here
if head == None:
return None
if k <= 0:
return None
p1, p2 = head, head
i = 1
while i < k:
i += 1
if p1.next:
p1 = p1.next
else:
return None
while p1.next:
p1 = p1.next
p2 = p2.next
return p2