1. 题目来源
链接:链表中倒数第k个节点
来源:LeetCode——《剑指-Offer》专项
2. 题目说明
输入一个链表,输出该链表中倒数第 k
个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第 1 个节点。例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6
。这个链表的倒数第 3 个节点是值为 4 的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
3. 题目解析
方法一:先走k - 1步解法
这个想法很简单,也只需要遍历一遍,但是针对链表的题坑点还是很多的:
head
为空,直接返回nullptr
- 链表节点数小于
k
时,采用这种解法会出错 - 若
k <= 0
会根据k
的数据类型,例如k
为无符号整形时,产生不可预计的错误
书上的坑点给列了这么三种,由于 LeetCode
有些不合法的条件都自行去掉了,但是作为一个优秀的程序员来讲,这些可能会导致代码崩溃的情况一定需要避免。
至于还有逆置链表的写法、有创建 vector<ListNde*>
的写法在此就不列了,效率都不怎么高,也不难,有兴趣可以自行实现。但是更为重要的还是要提升代码的鲁棒性。
参见代码如下:
// 执行用时 :0 ms, 在所有 C++ 提交中击败了100.00%的用户
// 内存消耗 :12.9 MB, 在所有 C++ 提交中击败了100.00%的用户
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
if (head == nullptr) return nullptr;
int count = 0;
ListNode* node = head;
ListNode* tmp = head;
while (head) {
head = head->next;
++count;
}
if (k > count) return nullptr;
while (k) {
tmp = tmp->next;
--k;
}
while (tmp) {
tmp = tmp->next;
node = node->next;
}
return node;
}
};