题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
使用双指针,第一个指针比第二个指针提前k步,当第一个指针到达链表结尾的时候,第二个指针就是指向结果。
答案
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
if(k < 0 || head == nullptr) return nullptr;
int tmp = 0;
ListNode* pre = head, *last = head;
while(tmp++ < k){
if(pre == nullptr) return nullptr;
pre = pre->next;
}
while(pre){
pre = pre->next;
last = last->next;
}
return last;
}
};