输入一个链表,输出该链表中倒数第k个结点。
思想有两种
一种是考虑到取倒数,可采用栈的形式先整体读入,然后再pop出栈第k个。
另一种是使用两个指针栈,两个指针相差k位,然后同时移动,直到后面的指针到达链表尾节点,此时前一个节点即是为需要求的倒数第k个结点。
(1)首先定义两个指针为p1,p2
(2)将p2先往前行走 k - 1 步(k-1才会得到倒数第k个)。操作时要注意是否出现k比链表长度还大。所以在行进过程中检查p2->next是否为空指针,是的话即返回空指针,否则前移。
for(unsigned int i=0; i<k-1; i++) { // 注意为k-1时才是倒数第k个
if(!pAhead->next) // 防止越界
return nullptr;
pAhead = pAhead->next;
}
(3)同时移动p1与p2,直到p2->next == nullptr,返回此时的p1即可。
总代码:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
if(!pListHead)
return nullptr;
ListNode* pNode = pListHead;
ListNode* pAhead = pListHead;
for(unsigned int i=0; i<k-1; i++) { // 注意为k-1时才是倒数第k个
if(!pAhead->next)
return nullptr;
pAhead = pAhead->next;
}
while(pAhead->next) {
pAhead = pAhead->next;
pNode = pNode->next;
}
return pNode;