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

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
1
2
3
4
5
6
7
8
9
10
11
12