剑指Offer-Python-链表中倒数第k个结点

题目:链表中倒数第k个结点

输入一个链表,输出该链表中倒数第k个结点。

思路1:一次遍历该链表,将结点一次存入列表res中,最后返回res[-k]。这里需要注意的是k的值是否符合题目要求,如果小于等于0或大于res长度则返回None。时间空间复杂度为O(n)。

代码:

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def FindKthToTail(self, head, k):
        # write code here
        if head is None:
            return head
        h = head
        res = []
        while h:
            res.append(h)
            h = h.next
        if k > len(res) or k <=0:
            return None
        return res[-k]

思路2:设置两个指针指向头结点。然后快指针往前移动k-1步(这是为了两个指针之间相差k)。最后快慢指针同时前进,当快指针到达末尾时,慢指针正好指向倒数k结点上。
代码

        if head is None or k <= 0:
            return None
        p = q = head
        i = k - 1
        while i > 0 and p:
            p = p.next
            i -= 1
        if p is None:
            return None
        while p.next:
            p = p.next
            q = q.next
        return q

猜你喜欢

转载自blog.csdn.net/qq_34364995/article/details/81254596