返回倒数第 k 个节点

概述:实现一种算法,找出单向链表中倒数第 k 个节点。返回该节点的值。

输入: 1->2->3->4->5 和 k = 2
输出: 4

方法一:双指针

思路:首先把快指针指向 k 位置,然后同步循环快慢指针,直到结束即可。

# 双指针
class Solution:
    def kthToLast(self, head: ListNode, k: int) -> int:
        pre, cur = head, head
        for i in range(k):
            cur = cur.next
        while cur:
            pre, cur = pre.next, cur.next
        return pre.val

方法二:快慢指针

思路:思路和上面完全一致,用 fast slow 来定于快慢指针。

# 快慢指针
class Solution:
    def kthToLast(self, head: ListNode, k: int) -> int:
        fast, slow = head, head
        while k > 0:
            fast = fast.next
            k -= 1
        while fast != None:
            fast = fast.next
            slow = slow.next
        return slow.val

方法三:哨点指针

思路:亦可以称为单指针,首先遍历算出链表的长度 n ,然后在 n - k 的长度中循环即可。 

# 哨点指针
class Solution:
    def kthToLast(self, head: ListNode, k: int) -> int:
        cur = head
        n = 0
        while cur != None:
            n += 1
            cur = cur.next
        cur = head
        for i in range(n - k):
            cur = cur.next
        return cur.val

总结

双指针妙啊!!!

Logo

聚焦前沿AI与大模型技术探索,汇聚开发者及爱好者,共享开源项目、学习资源与行业资讯。

更多推荐