本文最后更新于:2020年11月17日 晚上
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表:1->2->3->4->5,和k=2.
返回链表 4->5.
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def getKthFromEnd(self, head, k):
解题思路:
- 先遍历统计链表长度,记为n;
- 设置一个指针走(n-k)步,即可找到链表倒数第k个节点。
使用双指针则可以不同统计链表长度。
算法流程:
- 初始化:前指针$former$、后指针$latter$,双指针都指向头节点$head$。
- 构建双指针距离:前指针$former$先向前走$k$步(结束后,双指针$former$和$latter$间相距$k$步)。
- 双指针共同移动:循环中,双指针$former$和$latter$每轮都向前走一步,直到$former$走过链表尾节点时跳出(跳出后,$latter$与尾节点距离为$k-1$,即$latter$指向倒数第$k$个节点)。
- 返回值:返回$latter$即可。
复杂度分析:
- 时间复杂度$O(N)$:$N$为链表长度;总体看,$former$走了$N$步,$latter$走了$N-k$步。
- 空间复杂度$O(1)$:双指针$former$,$latter$使用常数大小的额外空间。
代码:
class Solution:
def getKthFromEnd(self, head, k):
former, latter = head, head
for _ in range(k):
former = former.next
while former:
former, latter = former.next, latter.next
本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!