【2.2链表 】链表中倒数第k个结点

题目描述

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

思路一

使用一个长度为k的尺子，pre先走k-1步，然后nex再开始走，当pre到达最后一个节点时，则nex所指向的就是倒是第k个结点

# -*- coding:utf-8 -*-
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
def FindKthToTail(self, head, k):
# 使用一个长度为k的尺子，pre先走k-1步，然后nex再开始走，
# 当pre到达最后一个节点时，则nex所指向的就是倒是第k个结点
if head == None or k < 1:
return None
pre = nex = head
i = 0
while i < k-1:
i += 1
if pre.next == None:
return None
else:
pre = pre.next

while pre.next != None:
pre = pre.next
nex = nex.next
return nex

思路二

把链表的node存入list中，然后返回list中倒数第k个元素

class Solution:
def FindKthToTail(self, head, k):
# 把链表的node存入list中，然后返回list中倒数第k个元素
if k < 1:
return None
node = []
while head != None:
node.append(head)
head = head.next
if len(node) >= k:
return node[-k]
else:
return None