剑指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;
}