输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
# 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:
# 这道题需要使用两个指针,指针1向后移动,到第k - 1个位置,这个时候第2个指针从链表的头部跟第1个指针同时向后移动,当第1个指针移动到链表的尾部时,第二个指针就到达了链表中倒数第k个节点
first_pointer = head
second_pointer = head
move_num = 0
while head.next:
head = head.next
# 当第1一个指针移动了k - 1个位置后,第二个指针也开始向后移动了
if move_num >= k - 1:
second_pointer = second_pointer.next
first_pointer = first_pointer.next
move_num += 1
# print(move_num)
if move_num < k - 1:
return None
else:
# 最后直接返回的是倒数第k个节点及之后的链表
return second_pointer