面试题22:链表中倒数第k个节点
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。
class Solution {
// 定义两个指针,
// 当第一个指针先走k-1步后第二个指针开始从头走,两个指针一起走,
// 当第一个指针走到最后时第二个指针指向倒数第k个节点;
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
if(head == nullptr || k==0)
return nullptr;
ListNode* p1Head = head;
ListNode* p2Head = nullptr;
for(int i=0;i<k-1;i++)
{
if(p1Head->next == nullptr)
return nullptr;
else
p1Head = p1Head->next;
}
p2Head = head;
while(p1Head->next != nullptr)
{
p1Head = p1Head->next;
p2Head = p2Head->next;
}
return p2Head;
}
};