快慢指针

快慢指针

快慢指针就是定义两根指针,移动的速度一快一慢,以此来制造出自己想要的差值。这个差值可以让我们找到链表上相应的节点。

剑指 Offer 22. 链表中倒数第k个节点

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。

例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

来源:力扣(LeetCode)

python代码

1
2
3
4
5
6
7
8
9
10
11
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
fast, slow = head, head
# 将第一个指针fast 指向链表的第 k + 1个节点
while fast and k > 0:
fast, k = fast.next, k - 1
# fast指针一直向后走到尾
while fast:
fast,slow = fast.next,slow.next
# 此时 slow 指针刚好距离fastk个节点
return slow

我们将第一个指针fast 指向链表的第 k + 1个节点,第二个指针 slow 指向链表的第一个节点,此时指针 fast 与 slow 二者之间刚好间隔 k 个节点。此时两个指针同步向后走,当第一个指针 fast 走到链表的尾部空节点时,则此时 slow 指针刚好指向链表的倒数第k个节点。