链表中倒数第k个节点

Posted 2021-01-01 嶙羽

题目描述

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

思路

注意，看清楚，是输出节点，而不是输出节点值

可以先求出链表总长度，然后正向遍历得到第n个节点

解答

class Solution:
    def FindKthToTail(self, head, k):
        count = 0
        p = head
        while p:
            count += 1
            p = p.next
        if count < k:
            return None
        number = count-k+1
        new_count = 0
        pHead = head
        while pHead:
            new_count += 1
            if new_count == number:
                return pHead
            pHead = pHead.next