题目:输入一个链表,输出该链表中倒数第K个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个结点。例如一个链表有6个结点,从头结点开始他们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。链表结点定义如下:
struct ListNode
{
int m_nValue;
ListNode* m_pNext;
};
思路:假设链表有n个结点,那么倒数第K个结点就是从头结点开始的第n-k+1个结点,这样需要遍历两遍链表,首先遍历整个链表为了得到链表中结点的个数n,然后遍历第二遍为了找到第n-k+1个结点,即倒数第K个结点。
只需要遍历一次的方法是:首先用一个指针指向头结点,然后开始往后走k-1步,然后用另一个指针指向头结点,这两个指针同时往后遍历,直到靠后的这个结点到达了尾结点,此时靠前的这个指针指向的是倒数第k个结点。也就是说,两个指针的距离保持在(k-2),然后同时遍历,这样靠后的结点是尾结点的时候,靠前的就是倒数第k个结点。
代码:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k)
{
if(pListHead == NULL || k == 0)
{
return NULL;
}
ListNode *pAhead = pListHead;
ListNode *pBehind = NULL;
for(unsigned int i = 0; i < k - 1; i++)
{
pAhead = pAhead->pNext;
if(pAhead == NULL)
return NULL;
}
pBehind = pListHead;
while(pAhead->pNext != NULL)
{
pAhead = pAhead->pNext;
pBehind = pBehind->pNext;
}
return pBehind;
}