LeetCode Algorithm 剑指 Offer 22. 链表中倒数第k个节点
题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
思路
快慢指针解法,先让fast指针走k步。当fast指针走k步后,fast指针和slow指针同步向后走,fast为None,slow就是倒数第k个位置。
代码
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
fast=head
slow=head
count=0
cur=head
while(fast):
if count >= k:
slow=slow.next
fast=fast.next
count+=1
return slow