1 问题
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围: 0 ≤ n ≤ 1 0 5 , 0 ≤ a i ≤ 1 0 9 , 0 ≤ k ≤ 1 0 9 0 \leq n \leq 10^5, 0 \leq a_i \leq 10^9, 0 \leq k \leq 10^9 0≤n≤105,0≤ai≤109,0≤k≤109
要求:空间复杂度 O(n),时间复杂度 O(n)
进阶:空间复杂度 O(1),时间复杂度 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
返回值:{}
2 答案
自己写的
class Solution:
def FindKthToTail(self , pHead: ListNode, k: int) -> ListNode:
n = 0
cur = pHead
while cur:
cur = cur.next
n += 1
m = n - k
if m < 0:
return cur # cur 就是空链表,也可以是None,或者直接写return
while m:
pHead = pHead.next
m -= 1
return pHead
官方解
- 快慢双指针
我们无法逆序遍历链表,就很难得到链表的倒数第k个元素,那我们可以试试反过来考虑,如果当前我们处于倒数第
k的位置上,即距离链表尾的距离是k,那我们假设双指针指向这两个位置,二者同步向前移动,当前面个指针到了链表头的时候,两个指针之间的距离还是k。虽然我们没有办法让指针逆向移动,但是我们刚刚这个思路却可以正向实施。
- step 1:准备一个快指针,从链表头开始,在链表上先走k步。
- step 2:准备慢指针指向原始链表头,代表当前元素,则慢指针与快指针之间的距离一直都是k。
- step 3:快慢指针同步移动,当快指针到达链表尾部的时候,慢指针正好到了倒数k个元素的位置。
class Solution:
def FindKthToTail(self , pHead , k ):
fast = pHead
slow = pHead
#快指针先行k步
for _ in range(0,k):
if fast != None:
fast = fast.next
#达不到k步说明链表过短,没有倒数k
else:
return None
#快慢指针同步,快指针先到底,慢指针指向倒数第k个
while fast:
fast = fast.next
slow = slow.next
return slow
- 先链表长度再找最后k,思路相同
链表不能逆向遍历,也不能直接访问。但是对于倒数第k个位置,我们只需要知道是正数多少位还是可以直接遍历得到的。
- step 1:可以先遍历一次链表找到链表的长度。
- step 2:然后比较链表长度是否比k小,如果比k小返回一个空节点。
- step 3:如果链表足够长,则我们从头节点往后遍历n−k次即可找到所求。
class Solution:
def FindKthToTail(self , pHead , k ):
n = 0
p = pHead
#统计链表长度
while p:
n = n + 1
p = p.next
#长度过小,返回空链表
if n < k:
return None
p = pHead
#遍历n-k次
for i in range(n-k):
p = p.next
return p
https://www.nowcoder.com/share/jump/9318638301698487863748