剑指offer 之 输出链表的倒数第K个结点

# -*- coding:utf-8 -*-
'''
输入一个链表，输出该链表中倒数第k个结点。
'''

'''
如果在只希望一次遍历的情况下, 寻找倒数第k个结点, 可以设置两个指针
第一个指针先往前走k-1步, 然后从第k步开始第二个指针指向头结点
然后两个指针一起遍历
当地一个指针指向尾节点的时候, 第二个指针正好指向倒数第k个结点
推广: 寻找中间节点, 两个指针一起, 第一个指针每次走两步, 第二个指针每次走一步,  快指针指到尾部, 慢指针正好指到中间
'''
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_ahead = head
		p_behind = None

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

		p_behind = head
		while p_ahead.next:
			p_ahead = p_ahead.next
			p_behind = p_behind.next

		return p_behind


node1 = ListNode(10)
node2 = ListNode(11)
node3 = ListNode(13)
node1.next = node2
node2.next = node3

S = Solution()
print(S.FindKthToTail(node1, 1).val)



# python 写法，遍历一遍取出list然后[-k]
# 此时输出的是倒数第k个结点的值，而不是结点
class Solution:
    def FindKthToTail(self, head, k):
        # write code here
        l=[]
        while head!=None:
            l.append(head)
            head=head.next
        if k>len(l) or k<1:
            return
        return l[-k]