问题重述:
题目:链表中倒数第K个结点
输入一个链表,求出链表中倒数第K个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第一个结点。例如一个链表有6个结点,从头开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。
思路解析:
有一种最直接的方法是将链表遍历两遍,第一遍得到链表长度,第二遍去找第K个结点。这种方式不免太麻烦。
我们可以设置两个指针Ahead和Behind,先将第Ahead遍历到从头开始的第K-1个结点处,然后另Behind置于头节点处,这样Behind就是相对于Ahead之处的倒数第K个结点,然后我们将两个指针一步一步向后移,这样当,Ahead移至链表末尾处的时候,Behind就是链表的到时第K个结点。
上图展示了求链表倒数第三个结点的过程。
代码实现:
在写代码的时候,我们要考虑代码的鲁棒性。
struct ListNode
{
int m_nvalue;
ListNode* m_pNext;
};
void InsertNode(ListNode *head, int value)
{
assert(head !=nullptr);
//构建新结点
ListNode *node = new ListNode;
node->m_nvalue = value;
node->m_pNext = nullptr;
//插入新结点
node->m_pNext = head->m_pNext;
head->m_pNext = node;
}
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k)
{
if (pListHead == nullptr || k == 0)
{
return nullptr;
}
ListNode* pAhead = pListHead;
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 = pListHead;
while (pAhead->m_pNext != nullptr)
{
pAhead=pAhead->m_pNext;
pBehind = pBehind->m_pNext;
}
return pBehind;
}
void Destroy(ListNode* head)
{
ListNode* temp = nullptr;
while (1)
{
temp = head->m_pNext;
if(temp!=nullptr)
{
head->m_pNext = temp->m_pNext;
free(temp);
temp = nullptr;
}
else
{
break;
}
}
free(head);
}
测试:
ListNode* head = new ListNode;
head->m_nvalue = 1;
head->m_pNext = nullptr;
InsertNode(head, 6);
InsertNode(head, 5);
InsertNode(head, 4);
InsertNode(head, 3);
InsertNode(head, 2);
ListNode* node = head;
while (node != nullptr)
{
cout << node->m_nvalue << ends;
node = node->m_pNext;
}
cout << endl;
cout << FindKthToTail(head, 3)->m_nvalue << endl;
Destroy(head);
结果: