题目描述:
输入一个链表,输出该链表中倒数第k个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个结点。例如一个链表有6个结点,从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。
解题思路:
为了能够只遍历一次就能找到倒数第k个节点,可以定义两个指针【这里又用到双指针,可见这种算法在链表系列中应用之广泛】:
1. 第一个指针从链表的头指针开始遍历向前走K-1,第二个指针保持不动;
2. 从第K步开始,第二个指针也开始从链表的头指针开始遍历;
3. 由于两个指针的距离保持在K-1,当第一个(走在前面的)指针到达链表的尾结点时,第二个指针(走在后面的)指针正好是倒数第K个结点。
Demo:
class Solution {
public:
ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
if (pListHead == nullptr || k == 0)
return nullptr;
ListNode *p1 = pListHead;
ListNode *p2 = pListHead;
for (int i = 0; i < k-1; ++i)
{
if (p1->next != nullptr)
p1 = p1->next;
else
return nullptr;
}
while (p1->next != nullptr)
{
p1 = p1->next;
p2 = p2->next;
}
return p2;
}
};