输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
例如,一个链表有 6
个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6
。这个链表的倒数第 3
个节点是值为 4
的节点。
这个问题就和曾经听过的一个问题很像,就是在只遍历链表一次的情况下拿到这个链表的中点,总的来说思路是差不多的。
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
result = head
tail = head
index = 1
while tail.next is not None:
if index == k:
tail = tail.next
result = result.next
continue
tail = tail.next
index += 1
return result
解题思路:
- 第一时间想到的解法:
- 先遍历统计链表长度,记为 n n n ;
- 设置一个指针走 ( n − k ) (n-k) (n−k) 步,即可找到链表倒数第 k k k 个节点。
- 使用双指针则可以不用统计链表长度。
算法流程:
- 初始化: 前指针
former
、后指针latter
,双指针都指向头节点 head 。 - 构建双指针距离: 前指针
former
先向前走 k k k 步(结束后,双指针former
和latter
间相距 k k k 步)。 - 双指针共同移动: 循环中,双指针
former
和latter
每轮都向前走一步,直至former
走过链表 尾节点 时跳出(跳出后,latter
与尾节点距离为 k − 1 k-1 k−1,即latter
指向倒数第 k k k 个节点)。 - 返回值: 返回
latter
即可。
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
former, latter = head, head
for _ in range(k):
former = former.next
while former:
former, latter = former.next, latter.next
return latter