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
上一篇下一篇

猜你喜欢

热点阅读