题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个节点是值为4的节点。
示例:
给定一个链表: 1->2->3->4->5, 和 k = 2.
返回链表 4->5.
思路
本题的基本思路是第一次遍历确定链表的节点数n,再进行一次遍历找到倒数第k个节点,它是顺序的第n-k+1个节点。如果k为1的话,就需要完整的遍历两次链表才能找到节点,这时就要考虑了,有没有什么办法可以一次就找到节点呢?我们可以再设置一个指针ptr指向当前链表的第一个节点head。首先将ptr向后移动k个位置,然后head和ptr同时向后移动,直到ptr指向nullptr。这就保证了程序永远只需对链表进行一次遍历。
代码
class Solution {
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
if(!head) return NULL;
ListNode* ptr=head;
while(k--){
if(!ptr) return NULL;
ptr=ptr->next;
}
while(ptr){
head=head->next;
ptr=ptr->next;
}
return head;
}
};