题目
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围:0 \leq n \leq 10^50≤n≤105,0 \leq a_i \leq 10^90≤ai≤109,0 \leq k \leq 10^90≤k≤109
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
进阶:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:
其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
示例1
输入:{1,2,3,4,5},2
复制返回值:{4,5}
复制说明:返回倒数第2个节点4,系统会打印后面所有的节点来比较。
示例2
输入:{2},8
复制返回值:{}
解题思路
1.遍历链表,并头插法到新链表中
2.指针移到链表最尾部,并统计节点个数
3.个数小于k,则返回空,反之取新链表前k位
题解
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
#
# @param pHead ListNode类
# @param k int整型
# @return ListNode类
#
class Solution:
def FindKthToTail(self , pHead: ListNode, k: int) -> ListNode:
if pHead is None: return None
# 1.遍历链表,并头插法到新链表中
new_pHead = pHead
#2. 指针移到链表最尾部,并统计节点个数
n = 1
while new_pHead.next:
n += 1
new_pHead = new_pHead.next
#3.个数小于k,则返回空,反之取新链表前k位
if n<k : return None
new_pHead = pHead
for i in range(n-k):
new_pHead = new_pHead.next
return new_pHead