跳转至

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

问题描述#

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

 

示例：

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

返回链表 4->5.

解题思路#

维护一个大小为 \(k\) 的队列，遍历完链表后，返回队列的第一个结点即可。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        # if head is None:
        #     return None

        vhead = ListNode(0)
        p, q, t = head, vhead, 0
        while p:
            while t >= k:
                vhead.next = vhead.next.next
                t -= 1
            q.next = p
            q = q.next
            p = p.next
            t += 1
        return vhead.next

快慢指针。

一个快指针 \(\text{fast}\) 先遍历 \(k\) 个结点，然后慢指针 \(\text{slow}\) 从头开始遍历，同时 \(\text{fast}\) 从停下来的下一个位置开始遍历，这样当 \(\text{fast}\) 指针遍历到为空时，\(\text{slow}\) 指针刚好指向倒数第 \(k\) 个结点。

1
2
3
4
5
6
7
8
9
class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        fast = head
        for _ in range(k):
            fast = fast.next
        slow = head
        while fast:
            slow, fast = slow.next, fast.next
        return slow
返回顶部

在手机上阅读