python-链表中倒数第k个节点
2020-03-14 本文已影响0人
JerryLoveCoding
leetcode: 输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
思路:既然是输出倒数第k的节点,那么可以设置2个指针,一个i指针一个j指针,刚开始都指向链表的头节点。
然后i指针先走k步,即i指针指向链表的第k个节点。然后i,j指针开始同步向后走,这样i指针走到头的时候,j指针也走到倒数第k个节点了。
代码:
# 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:
i, j = head, head
for _ in range(k):
if i: # i指针先走k步
i = i.next
while i:
i = i.next
j = j.next
return j