面试题15:链表中倒数第k个结点
一.题目描述
输入一个链表,输出该链表中倒数第k个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个结点。例如一个链表有6个结点,从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。
链表结点定义如下:
struct ListNode
{
int m_nValue;
ListNode* m_pNext;
}
二.分析问题
1.解题思路
假设整个链表有n个结点,那么倒数第k个结点就是从头结点开始的第n-k+1个结点。如果我们能够得到链表中结点的个数n,那我们只要从头结点开始往后走n-k+1步就可以了。如何得到结点数n?这个不难,只需要从头开始遍历链表,每经过一个结点,计数器加1就行了。也就是说我们需要遍历链表两次,第一次统计出链表中结点的个数,第二次就能找到倒数第k个结点。但我们希望解法只需要遍历链表一次。
为了实现只遍历链表一次就能找到倒数第k个结点,我们可以定义两个指针。第一个指针从链表的头指针开始遍历向前走k-1,第二个指针保持不动;从第k步开始,第二个指针也开始从链表的头指针开始遍历。由于两个指针的距离保持在k-1,当第一个(走在前面的)指针到达链表的尾结点时,第二个指针(走在后面的)指针正好是倒数第k个结点。
2.举例分析
下面以在有6个结点的链表中找倒数第3个结点为例分析这个思路的过程。首先用第一个指针从头结点开始向前走两(2=3-1)步到达第3个结点。接着我们把第二个指针初始化指向链表的第一个结点。最后让两个指针同时向前遍历,当第一个指针到达链表的尾结点时,第二个指针指向的刚好就是倒数第3个结点。
注:
(a)第一个指针在链表上走两步。
(b)把第二个指针指向链表的头结点。
(c)两个指针一同沿着链表向前走。当第一个指针指向链表的尾结点时,第二个指针指向倒数第3个结点。
三.代码
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k)
{
if (pListHead == NULL || k == 0)
{
return NULL;
}
ListNode* p1 = pListHead;
ListNode* p2 = NULL;
for (unsigned int i = 0; i < k - 1; ++i)
{
if (p1->m_pNext != NULL)
{
p1 = p1->m_pNext;
}
else
{
return NULL;
}
}
p2 = pListHead;
while (p1->m_pNext != NULL)
{
p1 = p1->m_pNext;
p2 = p2->m_pNext;
}
return p2;
}