【剑指 Offer】22,链表中倒数第k个节点。 难度等级:简单。思路:快慢指针

1. 题目

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。

例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

示例:

给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.

2. 我的解法:遍历两次链表

思路:

第一次遍历链表,获得链表长度;
第二次遍历链表,找到第 length-k 个节点并返回。

code:

# 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:
        # 第一次遍历链表,获得链表长度
        length=1
        tail=head
        while tail:
            tail=tail.next
            length+=1

        # 第二次遍历链表,找到第 length-k 个节点并返回
        start=1
        while start!=length-k:
            head=head.next
            start+=1
        return head

分析:时间复杂度为 O(n),空间复杂度为 O(1)

3. 更优解法:快慢指针,遍历一次链表

思路:快慢指针

首先让 slow 指向头节点,fast 指向第 k+1 个节点;
然后让 slow 和 fast 同步向后走,当 fast 指向链表的尾部空节点时,返回 slow 即可。

code:

# 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:
        # 快慢指针:始终让 fast 指针领先 slow 指针 k 个节点
        slow=head
        fast=head
        for _ in range(k):
            fast=fast.next
        
        # 当 fast = None 时返回 slow
        while fast:
            slow=slow.next
            fast=fast.next
        return slow

分析:时间复杂度为 O(n),空间复杂度为 O(1)

猜你喜欢

转载自blog.csdn.net/qq_43799400/article/details/131114538