140. LCR 训练计划 II

LCR 140. 训练计划 II

题目：

给定一个头节点为 head 的链表用于记录一系列核心肌群训练项目编号，请查找并返回倒数第 cnt 个训练项目编号。

示例：

示例 1：

输入：head = [2,4,7,8], cnt = 1
输出：8

提示：

• 1 <= head.length <= 100
• 0 <= head[i] <= 100
• 1 <= cnt <= head.length

解题：

方法一：双指针法

1. 初始化： 前指针 first 、后指针 last，双指针都指向头节点 head 。

2. 构建双指针距离： 前指针 first 先向前走 cnt 步（结束后，双指针 first 和 last 间相距 cnt 步）。

3. 双指针共同移动： 循环中，双指针 first 和 last 每轮都向前走一步，直至 first 走过链表 尾节点 时跳出（跳出后，last 与尾节点距离为 cnt?1 ，即 last 指向倒数第 cnt 个节点）。

4. 返回值： 返回 last 即可。

class Solution {
public:
    ListNode* trainingPlan(ListNode* head, int cnt) {
        ListNode* first = head, *last = head;
        for(int i = 0; i < cnt; ++i)
            first = first->next;
        while(first != nullptr) {
            first = first->next;
            last = last->next;
        }
        return last;
    }
};

本题没有 cnt > 链表长度的测试样例 ，因此不用考虑越界。考虑越界问题的代码如下：

class Solution {
public:
    ListNode* trainingPlan(ListNode* head, int cnt) {
        ListNode *former = head, *latter = head;
        for(int i = 0; i < cnt; i++) {
            if(former == nullptr) return nullptr;
            former = former->next;
        }
        while(former != nullptr) {
            former = former->next;
            latter = latter->next;
        }
        return latter;
    }
};

复杂度分析：

• 时间复杂度 O(n) ： n 为链表长度；总体看，first 走了 n 步，last走了 (?cnt) 步。
• 空间复杂度 O(1) ： 双指针 first , last 使用常数大小的额外空间。