美文网首页
python-链表中倒数第k个节点

python-链表中倒数第k个节点

作者: JerryLoveCoding | 来源:发表于2020-03-14 18:30 被阅读0次

    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
    

    相关文章

      网友评论

          本文标题:python-链表中倒数第k个节点

          本文链接:https://www.haomeiwen.com/subject/avngshtx.html