剑指offer JZ14:链表中倒数第k个结点

剑指offer JZ14:链表中倒数第k个结点

问题

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

思路

首先,看到问题的第一反应就是将链表的每个结点按顺序输出到列表里,再将列表中的倒数第k项输出即可,但是需要消耗0(N)的空间。

这里运用一个非常巧妙的双指针算法:要输出倒数第k项,只需要让一个指针先跑k-1项,再让两个指针同时向后移动,当第一个指针到达队尾时,第二个指针指向的自然就是倒数第k项了。

注意对于k>n等特殊情况的讨论

代码及解释

class Solution:
    def FindKthToTail(self, head, k):
        # write code here
        if head == None:
            return None
        if k <= 0:
            return None
        p1, p2 = head, head
        i = 1
        while i < k:
            i += 1
            if p1.next:
                p1 = p1.next
            else:
                return None
        while p1.next:
            p1 = p1.next
            p2 = p2.next
        return p2

你可能感兴趣的:(剑指offer)