剑指 Offer 22. 链表中倒数第k个节点
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
方法:
(1)初始化: 前指针 first、后指针 second ,双指针都指向头节点 head 。
(2)构建双指针距离: 前指针 first 先向前走 k 步(结束后,双指针 first 和 second 间相距 k 步)。
(3)双指针共同移动: 循环中,双指针 first 和 second 每轮都向前走一步,直至 first 走过链表 尾节点 时跳出(跳出后, second 与尾节点距离为 k−1,即 latter 指向倒数第 k 个节点)。
(4)返回值: 返回 second 即可。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
first, second = head, head
for _ in range(k):
if not first:
return
first = first.next
while first:
first, second = first.next, second.next
return second