题目:输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个节点。例如,一个链表有6个节点,从头节点开始,它们的值依次是1,2,3,4,5,6。这个链表的倒数第3个节点是值为4的节点。
链表定义如下:
struct ListNode
{
int value;
ListNode* next;
};
分析:为了得到倒数第k个节点,很自然的想法是先走到链表的尾端,再从尾端回溯k步,但是单向链表的结点只有从前往后的指针,没有从后向前的指针,因此这种思路行不通。
思路1:假设整个链表有n个节点,那么倒数第k个节点就是从头节点开始的第n-k+1个节点,如果能得到链表中节点的个数n,那么只要从头节点开始往后走n-k+1步就可得到第倒数k个节点,,如何得到节点数n?从头遍历开始遍历链表,每经过一个节点,计数器加1。但是这种方法需要遍历两次链表,第一次统计链表中节点的个数,第二次找到倒数第k个节点。
思路2:定义两个指针,第一个指针从链表的头指针开始遍历向前走k-1步,第二个指针保持不动,从第k步开始,第二个指针也开始从链表的头指针开始遍历。由于两个指针的距离保持在k-1,当第一个(走在前面的)指针到达链表的未节点时,第二个(走在后面的)指针正好指向倒数第k个节点。
下面以在有6个节点的链表中找倒数第3个节点为例,按照思路2进行分析:
(1)第一个指针p1从头节点开始向前走2(2=k-1=3-1)步到达第3个节点,如图1所示;
(2)将第二个指针p2指向链表的第一个节点,如图2所示;
(3)让两个指针同时向前遍历,当第一个指针p1到达链表的尾节点时,第二个指针p2指向的结点正好是倒数第3个节点,如图3所示。
具体代码如下图所示:
#include <iostream>
#include <assert.h>
using namespace std;
struct ListNode
{
int value;
ListNode* next;
};
void Init(ListNode* pListHead)
{
assert(pListHead != NULL);
if(pListHead == NULL)
{
return ;
}
pListHead->next=NULL;
}
static ListNode* Buynode()
{
ListNode *pnewnode=new ListNode();
assert(pnewnode != NULL);
pnewnode->next=NULL;
return pnewnode;
}
bool InsertTail(ListNode *pListHead,int val)
{
if(pListHead == NULL)
{
return false;
}
ListNode *pCur=pListHead;
while(pCur->next != NULL)
{
pCur=pCur->next;
}
ListNode *pnewnode=Buynode();
pnewnode->value=val;
pCur->next=pnewnode;
return true;
}
ListNode* FindKthToTail(ListNode *pListHead,unsigned int k)
{
if(pListHead == NULL || k == 0)
{
return NULL;
}
ListNode *phead=pListHead;
ListNode *pbehind=NULL;
for(unsigned int i=0;i<k-1;++i)
{
if(phead->next != NULL)
{
phead=phead->next;
}
else
{
return NULL;
}
}
pbehind=pListHead;
while(phead->next != NULL)
{
phead=phead->next;
pbehind=pbehind->next;
}
return pbehind;
}
void Print(ListNode* pListHead)
{
if(pListHead == NULL)
{
return;
}
struct ListNode* pCur=pListHead->next;
while(pCur != NULL)
{
printf("%d ",pCur->value);
pCur=pCur->next;
}
printf("\n");
}
void Clear(ListNode* pListHead)
{
struct ListNode* pCur=pListHead->next;
struct ListNode* pNext=pCur;
while(pCur != NULL)
{
pNext=pCur->next;
free(pCur);
pCur=pNext;
}
pListHead->next=NULL;
}
void Destory(ListNode* pListHead)
{
Clear(pListHead);
}
int main()
{
ListNode pListHead;
Init(&pListHead);
for(int i=0;i<6;i++)
{
InsertTail(&pListHead,i+1);
}
Print(&pListHead);
ListNode *tmp=NULL;
tmp=FindKthToTail(&pListHead,3);
cout<<tmp->value<<endl;
tmp=FindKthToTail(&pListHead,1);
cout<<tmp->value<<endl;
tmp=FindKthToTail(&pListHead,6);
cout<<tmp->value<<endl;
Destory(&pListHead);
Print(&pListHead);
return 0;
}
运行结果如下图所示,下图分别打印了倒数第3个节点的值,倒数第1个节点的值和倒数第六个结点的值。