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

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。

示例:

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

解法

  • 先遍历一遍获得长度n,然后再遍历到n-k即可得到节点
class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        total = 0
        cur = head
        while cur:
            cur = cur.next
            total += 1
        n = total - k
        node = head
        while n:
            node = node.next
            n -= 1
        return node
  • 双指针法。创建两个指针fast, slow,快指针先走k个,然后快指针和慢指针一起走,当快指针为空的时候,慢指针的位置即为结果。
class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        fast, slow = head,head
        while k:
            fast = fast.next
            k -= 1
        while fast:
            fast,slow = fast.next,slow.next
        return slow