题目
输入一个链表,输出该链表中倒数第 k k k个结点。
注意:
k >= 1
;- 如果 k k k 大于链表长度,则返回$ NULL$;
样例
输入:链表:1->2->3->4->5 ,k=2
输出:4
思路
(链表)
O
(
n
)
O(n)
O(n)
由于单链表不能索引到前驱节点,所以只能从前往后遍历。
我们一共遍历两次:
第一次遍历得到链表总长度
n
n
n;
链表的倒数第
k
k
k 个节点,相当于正数第
n
−
k
+
1
n−k+1
n−k+1 个节点。所以第二次遍历到第$ n−k+1$ 个节点,就是我们要找的答案。
注意当
k
>
n
k>n
k>n 时要返回
n
u
l
l
p
t
r
nullptr
nullptr。
时间复杂度
链表总共遍历两次,所以时间复杂度是
O
(
n
)
O(n)
O(n)。
代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* findKthToTail(ListNode* head, int k) {
int n = 0;
for(auto p = head; p ; p = p->next) n++;
if(k > n) return NULL;
auto p = head;
for(int i = 0; i < n - k; i++) p = p->next;
return p;
}
};