题目来自剑指Offer
题目
代码
#include <iostream>
#include <assert.h>
using namespace std;
struct ListNode
{
int m_Data;
ListNode* m_pNext;
};
/*
注意初始化时,快指针应指向第k个结点,慢指针指向第1个结点。
*/
ListNode* FindKthToTail(ListNode* pListHead,unsigned int k)
{
assert(pListHead != NULL && k != 0);
unsigned int i = 1;
ListNode* pFirstNode = pListHead;
ListNode* pSecNode = pListHead;
while (pFirstNode && i <= k)
{
pFirstNode = pFirstNode->m_pNext;
i++;
}
if (i <= k)
{
return NULL;
}
while (pFirstNode)
{
pSecNode = pSecNode->m_pNext;
pFirstNode = pFirstNode ->m_pNext;
}
return pSecNode;
}
void CreateList(ListNode** pHead,int nLen)//头指针使用指针的指针
{
assert(*pHead == NULL && nLen > 0);
ListNode* pCur = NULL;
ListNode* pNewNode = NULL;
for (int i = 0;i < nLen;i++)
{
pNewNode = new ListNode;
cin>>pNewNode->m_Data;
pNewNode->m_pNext = NULL;
if (*pHead == NULL)
{
*pHead = pNewNode;
pCur = *pHead;
}
else
{
pCur->m_pNext = pNewNode;
pCur = pNewNode;
}
}
}
void PrintList(ListNode* pCurNode)
{
assert(pCurNode != NULL);
while (pCurNode)
{
cout<<pCurNode->m_Data<<" ";
pCurNode = pCurNode->m_pNext;
}
cout<<endl;
}
int main()
{
int nLen = 0;
ListNode* pHead = NULL;
cout<<"please input node num: ";
cin >> nLen;
CreateList(&pHead,nLen);//注意传参使用引用
PrintList(pHead);
ListNode* pKthToTail = FindKthToTail(pHead,1);
if (pKthToTail)
{
cout<<"倒数第一个结点:"<<pKthToTail->m_Data<<endl;
}
else
{
cout<<"倒数第一个结点不存在!"<<endl;
}
pKthToTail = FindKthToTail(pHead,5);
if (pKthToTail)
{
cout<<"倒数第五个结点:"<<pKthToTail->m_Data<<endl;
}
else
{
cout<<"倒数第五个结点不存在!"<<endl;
}
pKthToTail = FindKthToTail(pHead,3);
if (pKthToTail)
{
cout<<"倒数第三个结点: "<<pKthToTail->m_Data<<endl;
}
else
{
cout<<"倒数第三个结点不存在!"<<endl;
}
system("pause");
return 1;
}
注意:
(1)链表中没有k个结点的情况
(2)初始化时慢指针指向第一个结点,快指针指向第k个结点。