牛客 BM8 链表倒数最后k个结点

题目

输入一个长度为 n 的链表，设链表中的元素的值为 ai ，返回该链表中倒数第k个节点。
如果该链表长度小于k，请返回一个长度为 0 的链表。

例如输入{1,2,3,4,5},2时，对应的链表结构如下图所示：

其中蓝色部分为该链表的最后2个结点，所以返回倒数第2个结点（也即结点值为4的结点）即可。

示例

输入：{1,2,3,4,5},2
返回值：{4,5}
说明：返回倒数第2个节点4，系统会打印后面所有的节点来比较。 

我的题解

1、先遍历链表，得出链表的长度
2、根据链表长度判断是不是倒数第k个节点，是就返回，不是就继续遍历

class Solution:
    def FindKthToTail(self , pHead: ListNode, k: int) -> ListNode:
        # write code here
        h = pHead
        len = 0
        while h != None:
            len += 1
            h = h.next
        if len < k:
            return h
        l = pHead
        while len > k:
            len -= 1
            l = l.next
        return l

使用的是笨方法，有更优解的欢迎指正！