寻找链表的倒数第K个节点

题目：输入一个单向链表，输出该链表中倒数第k 个结点。链表的倒数第0 个结点为链表的尾指针。

链表数据结构定义如下：

struct LISTNODE
{
    int nData;

    LISTNODE *next;
};

 

思路：用两个指针，第一个指针先走k步，然后第二个指针指向第一个节点，开始和第一个指针一起往后走。

注意：1、我假设这个链表有头节点，若要求链表无头节点，则代码需做修改，但核心思想是一样的

　　　2、注意k的范围，对于k的各种取之给一个合适的反馈。

 

基于以上分析，写出代码：

LISTNODE *GetLastKNode(LISTNODE *head, int k)
{
    assert (head != NULL);

    assert (k >= 0);

    LISTNODE *pPre = head->next;

    if (NULL == pPre)
    {
        cerr << "空链表" << endl << endl;
        exit (1);
    }

    if (0 == k)
    {
        cout << "链表倒数第 " << k << " 个节点是尾指针.\n" << endl;
        exit (0);
    }

    while ((pPre != NULL) && (k > 0))
    {
        pPre = pPre->next;
        --k;
    }

    if (k > 0)
    {
        cout << "K的值大于链表节点的个数.\n\n";
        exit (1);
    }

    LISTNODE *pLast = head->next;

    while (pPre != NULL)
    {
        pPre = pPre->next;
        pLast = pLast->next;
    }

    return (pLast);
}

 

　　下面是一些测试结果：

k的取之很合理的时候：

k取值为0的时候：

K的取值不合理的时候：

链表为空的时候：