描述
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围:
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
进阶:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:
其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
示例1
输入:{1,2,3,4,5},2
返回值:{4,5}
说明:返回倒数第2个节点4,系统会打印后面所有的节点来比较。
示例2
输入:{2},8
返回值:{}
方法一:数组法
把每个节点装进数组中,访问倒数第 k 个元素。
class Solution {
public:
ListNode* FindKthToTail(ListNode* pHead, int k) {
// write code here
vector<ListNode*> ret;
for( ; pHead != NULL; pHead = pHead->next)
ret.emplace_back(pHead); // 把每个节点指针放入数组
if(k > ret.size() || !k) return nullptr; // 判断非法的 k 值
return ret[ret.size() - k];
}
};
运行时间:31ms
超过2.62% 用C++提交的代码
占用内存:6576KB
超过1.94%用C++提交的代码
方法二:双指针法
- 用两个指针 left, right 指向头节点
- 把 right 指针后移 k 个位置
- 把 left, right 同时后移直到 right 到链表尾部,此时 left 指向的就是倒数第 k 个节点。
ListNode* FindKthToTail(ListNode* pHead, int k) {
ListNode* right = pHead;
while(k-- && right != nullptr){
right = right->next; // 移动右侧指针造成 k 的距离差
}
if(k >= 0) return nullptr;// 此时说明 k 比链表长度长
ListNode* left = pHead;
while (right != nullptr){
right = right->next;
left = left->next; // 两个指针一起移动找到倒数第 k 个节点
}
return left;
}
运行时间:26ms
超过13.59% 用C++提交的代码
占用内存:5152KB
超过29.81%用C++提交的代码