题目:输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第一个节点。例如一个链表有6个节点,从头开始他们的值依次是1、2、3、4、5、6。这个链表的倒数第三个节点是值为4的节点。链表的数据结构如下:
struct ListNode {
int m_nValue;
ListNode *m_pNext;
};
思路:
双指针“追赶”法,注意链表节点数小于输入参数k的情况。
编译环境:ArchLinux+Clang3.3,C++11
实现一:
#include <iostream>
#include <cassert>
using namespace std;
struct ListNode {
int m_nValue;
ListNode *m_pNext;
};
ListNode *FindKthToTail(ListNode *pListHead, unsigned k)
{
if (pListHead == nullptr || k == 0) return nullptr;
ListNode *pAhead = pListHead;
ListNode *pBehind = pListHead;
unsigned remained = k;
while (remained && pAhead) { // pAhead先至多移动k步
pAhead = pAhead->m_pNext;
remained--;
}
if (remained)
return nullptr;
while (pAhead) { // pAhead - pBehiend = k
pAhead = pAhead->m_pNext;
pBehind = pBehind->m_pNext;
}
return pBehind;
}
void printList(ListNode *pListHead)
{
ListNode *pNode = pListHead;
while (pNode) {
cout << pNode->m_nValue << '\t';
}
if (pListHead) {
cout << endl;
}
}
/** 测试案例 **/
int main()
{
/** 构造链表{5,4,3,2,1} **/
int a[]{1,2,3,4,5};
ListNode *pListHead = nullptr;
for (auto i : a) {
pListHead = new ListNode{i, pListHead};
}
/** 测试 **/
assert(FindKthToTail(pListHead, 1)->m_nValue == 1);
assert(FindKthToTail(pListHead, 5)->m_nValue == 5);
assert(FindKthToTail(pListHead, 6) == nullptr);
assert(FindKthToTail(nullptr, 6) == nullptr);
assert(FindKthToTail(pListHead, 0) == nullptr);
cout << "OK!" << endl;
}