链表中倒数第K个节点
1、题目
输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个节点,该链表是单向链表,不存在指向前的指针。
输入参数:一个指向链表头节点的指针pHead,一个无符号整数k
输出结果:返回倒数第k个节点,或者返回空指针
2、解题
这道题的关键在于快慢指针的使用。
普通思路是遍历两遍链表,第一遍获得节点个数n,然后获得倒数第k个节点的位置n-k+1,第二遍直接遍历到节点
好的思路是采用快慢指针的解法,先让快指针走k-1步,然后慢指针和快指针同时走,快指针到达链表尾结点结束
这道题的坑在于鲁棒性的检查
首先要对传入参数进行检查,然后在快指针使用for循环前进时也要时刻检查空指针的问题,这两步走完,后面慢指针就不需要考虑空指针的问题了,因为之前的路径是已经被验证存在的。
3、代码
ListNode* FindKthToTail(ListNode* pHead, unsigned int k) {
//鲁棒性检查
if (pHead == nullptr || k == 0)
return nullptr;
ListNode* pAhead = pHead;
ListNode* pBehind = nullptr;
for (unsigned int i = 0; i < k - 1; i++) {
if (pAhead->m_pNext != nullptr) {
pAhead = pAhead->m_pNext;
}
else {
return nullptr;
}
}
pBehind = pHead;
while (pAhead->m_pNext != nullptr) {
pAhead = pAhead->m_pNext;
pBehind = pBehind->m_pNext;
}
return pBehind;
}
4、注意点
- 参数k和for循环使用的变量i使用的是无符号整数类型
- 有一个细节,慢指针pBehind在快指针for循环后才赋初值,在出现空指针异常时,节省了指向头节点的时间