剑指Offer算法实现之十五：链表中倒数第k个结点

题目：输入一个链表，输出该链表中倒数第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;}