今日主要总结一下,剑指 Offer 22. 链表中倒数第k个节点
题目:剑指 Offer 22. 链表中倒数第k个节点
题目描述:
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回NULL!!!
示例 1:
输入:{1,2,3,4,5},2
返回值:{4,5}
说明:返回倒数第2个节点4,系统会打印后面所有的节点来比较。
其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
示例 2:
输入:{2},8
返回值:{}
要求:空间复杂度 O(n),时间复杂度 O(n)
进阶:空间复杂度 O(1),时间复杂度 O(n)
本题重难点
1.这道题要考虑如果该链表长度小于k,请返回NULL的情况
2.进阶:是否可以使用 O(1) 空间解决此题?
一、栈
这道题比较容易想到的就是使用栈把所有节点都存下,之后把(k-1)个栈顶元素pop弹出栈,再return 的栈顶元素就是所求节点
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
stack<ListNode*> st;
int n = 0;
while(head){
st.push(head);
head = head->next;
++n;
}
if(k > n)
return NULL;
for(int i = 0; i < k - 1; ++i){
st.pop();
}
return st.top();
}
};
复杂度分析
时间复杂度:O(N)
空间复杂度:O(N) 因为使用了额外栈空间来存储链表节点
二、顺序查找
思路与算法
最简单直接的方法即为顺序查找,假设当前链表的长度为 n,则我们知道链表的倒数第 k 个节点即为正数第 n - k 个节点,此时我们只需要顺序遍历到链表的第 n - k 个节点即为倒数第 k 个节点。
我们首先求出链表的长度 n,然后顺序遍历到链表的第 n - k 个节点返回即可。
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
int n = 0;
ListNode* cur = head;
while(cur){
++n;
cur = cur->next;
}
if(k > n)
return NULL;
for(int i = 0; i < n - k; ++i){
head = head->next;
}
return head;
}
};
复杂度分析
时间复杂度:O(N)
空间复杂度:O(1)
三、双指针
算法流程:
- 初始化: 前指针 former 、后指针 latter ,双指针都指向头节点 head
- 构建双指针距离: 前指针 former 先向前走 k 步(结束后,双指针 former 和 latter 间相距 k 步)。
- 双指针共同移动: 循环中,双指针 former 和 latter 每轮都向前走一步,直至 former 走过链表 尾节点 时跳出(跳出后, latter 与尾节点距离为 k-1,即 latter 指向倒数第 k个节点)。
- 返回值: 返回 latter 即可。
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
ListNode* slow = head;
ListNode* fast = head;
for(int i = 0; i < k; ++i){
if(fast == NULL)
return NULL;
fast = fast->next;
}
while(fast){
fast = fast->next;
slow = slow->next;
}
return slow;
}
};
复杂度分析
时间复杂度:O(N)
空间复杂度:O(1) 我们只使用了slow,fast两个指针
总结
二、算法思想
链表题,要想到使用双指针的方法
注意!!!
当使用fast 指针向后移动两个位置需要多加一层if判断fast -> next 是否为NULL