文章目录
【面试题22 链表中倒数第k个节点】
难度: 简单
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
Leetcode题目对应位置: 面试题22:链表中倒数第k个节点
常规解法是先遍历一遍单链表,求得链表长度,此时就能得到倒数第 k 个节点位置。再次遍历单链表,从计算出的输出位置进行打印即可。时间复杂度:O(n),需要遍历两次链表。空间复杂度:O(1)。这里就不写啦。
思路:双指针
双指针法: 不需要统计链表长度,只需要维护两个指针,让两个指针中间间隔 k-1
个节点,即快指针先走 k
步。然后两个指针再一起走,当快指针指向为空时,慢指针所指就是倒数第 k 个节点。
时间复杂度: O(n),仅需遍历单链表一遍
空间复杂度: O(1),仅需维护两个指针 i
,j
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
i, j, s = head, head, 0
while s < k:
j = j.next
s += 1
while j:
i, j = i.next, j.next
return i
特别地,还可以省掉第一个 while 循环:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
i, j, s = head, head, 0
while j:
if s >= k: i = i.next
j = j.next
s += 1
return i
总结一下: 每次看到这种一眼看上去貌似需要遍历两次链表的题,要下意识思考有没有时间复杂度更优的方法!