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
解题:
方法一:双指针法
-
初始化: 前指针 first 、后指针 last,双指针都指向头节点 head 。
-
构建双指针距离: 前指针 first 先向前走 cnt 步(结束后,双指针 first 和 last 间相距 cnt 步)。
-
双指针共同移动: 循环中,双指针 first 和 last 每轮都向前走一步,直至 first 走过链表 尾节点 时跳出(跳出后,last 与尾节点距离为 cnt−1 ,即 last 指向倒数第 cnt 个节点)。
-
返回值: 返回 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 使用常数大小的额外空间。