题目
输入一个链表,输出该链表中倒数第k个结点。
解决思路
定义两个指针。第一个指针从链表的开始遍历,向前走k-1步,第二个指针保持不动。从第k步开始,第二个指针也开始从链表的头指针开始遍历。由于两个指针的距离保持在k-1,当第一个指针到达链表的尾结点时,第二个指针正好指向倒数第k个结点。
代码
#include <iostream>
struct ListNode
{
int m_value;
ListNode* m_pNext;
};
// 查找链表中倒数第K个结点
ListNode* FindKthToTail(ListNode* pListHead, unsigned k)
{
// 头结点不能为空,且K不能为0
if (pListHead == nullptr || k == 0)
return nullptr;
// pFront:前面先走k-1步的结点
// pBehind:后面的结点
ListNode *pFront = pListHead;
ListNode *pBehind = nullptr;
// pBehind与pFront相差k-1个结点
for (unsigned i = 0; i < k - 1; ++i) {
// 如果链表的总结点数少于K,则返回空指针
if (pFront->m_pNext != nullptr)
pFront = pFront->m_pNext;
else
return nullptr;
}
pBehind = pListHead;
// 当pFront走到链表末尾,则pBehind就是倒数第K个结点
while (pFront->m_pNext != nullptr)
{
pFront = pFront->m_pNext;
pBehind = pBehind->m_pNext;
}
return pBehind;
}
int main()
{
return EXIT_SUCCESS;
}
举一反三
当我们用一个指针遍历链表不能解决问题的时候,可以尝试用两个指针来遍历链表。可以让其中一个指针遍历的速度快一些(比如一次在链表中走两步),或者让它先在链表上走若干步。
如:求链表的中间结点。
如果链表中结点为奇数,返回中间结点;如果结点总数是偶数,返回中间两个结点的任意一个。可以定义两个指针,同时从链表的头结点出发,一个指针一次走一步,另一个指针一次走两步。当走得快的指针走到链表的末尾时,走得慢的指针正好在链表的中间。
判断一个链表是否是形成了环形结构。定义两个指针,同时从链表的头结点出发,一个指针一次走一步,另一个指针一次走两步。如果走得快的指针追上了走得慢的指针,那么链表就是环形链表;如果走得快的指针走到了链表的末尾都没追上第一个指针,那么链表就不是环形链表。