Leetcode刷题:剑指offer【面试题22 链表中倒数第k个节点】

文章目录

  • 思路:双指针

【面试题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),仅需维护两个指针 ij

# 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

总结一下: 每次看到这种一眼看上去貌似需要遍历两次链表的题,要下意识思考有没有时间复杂度更优的方法!

你可能感兴趣的:(今天刷题了吗)