刷题--链表中倒数第k个结点

输入一个链表,输出该链表中倒数第k个结点。
基本思想:定义两个指针。第一个指针从链表的头结点开始遍历向前走k-1,第二个指针保持不动;从第k步开始,第二个指针也开始从链表的头结点开始遍历。由于两个指针的距离保持在k-1,当第一个指针到达链表的尾结点时,第二个指针正好是倒数第k个结点。
考虑链表为空,链表结点总数少于k,k非正。

# -*- 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 == None or k <= 0:
            return None

        p = head
        for i in range(k - 1):
            if p.next:
                p = p.next
            else:
                return None

        q = head
        while p.next:
            p = p.next
            q = q.next

        return q

猜你喜欢

转载自blog.csdn.net/treasure_z/article/details/80015040