给定一个头节点为 head 的链表用于记录一系列核心肌群训练项目编号,请查找并返回倒数第 cnt 个训练项目编号。
示例:
输入:head = [2,4,7,8], cnt = 1
输出:8
算法思想
先遍历统计链表长度,记为 n ;
设置一个指针走 (n−cnt) 步,即可找到链表倒数第cnt 个节点;
使用双指针则可以不用统计链表长度。
初始化: 前指针 former 、后指针 latter ,双指针都指向头节点 head 。
构建双指针距离: 前指针 former 先向前走 cnt步(结束后,双指针 former 和 latter 间相距 cnt 步)。
双指针共同移动: 循环中,双指针 former 和 latter 每轮都向前走一步,直至 former 走过链表 尾节点 时跳出(跳出后,latter 与尾节点距离为 cnt−1,即 latter 指向倒数第 cnt个节点)。
返回值: 返回 latter 即可。
class Solution:
def trainingPlan(self, head: ListNode, cnt: int) -> ListNode:
former, latter = head, head
for _ in range(cnt):
former = former.next
while former:
former, latter = former.next, latter.next
return latter
时间复杂度 :O(n)
空间复杂度 :O(1)