ACM. HJ51 输出单向链表中倒数第k个结点 ●

HJ51 输出单向链表中倒数第k个结点 ●

描述

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

链表结点定义如下：

struct ListNode
{
    int m_nKey;
    ListNode* m_pNext;
};

正常返回倒数第 k 个结点指针，异常返回空指针.

要求：
(1)正序构建链表;
(2)构建后要忘记链表长度。
数据范围：链表长度满足 $1\le n\le 1000$， $k\le n$，链表中数据满足 $0\le val\le 10000$

本题有多组样例输入。

输入描述：

输入说明

1. 输入链表结点个数
2. 输入链表的值
3. 输入 k 的值

输出描述：

输出一个整数

示例

输入：
8
1 2 3 4 5 6 7 8
4
输出：
5

题解

1. 快慢指针

多组测试用例时，每次循环内输出 cout 即可。

利用快慢指针进行倒数第 k 个数的判断：
快指针比慢指针快 k 步，当快指针走到 null 节点时，慢指针此时指向倒数第 k 个数。

#include <iostream>
#include <vector>
using namespace std;

struct ListNode
{
    int m_nKey;
    ListNode* m_pNext;
};

int main(){
    int n, k, num;
    while(cin >> n){        // 多组用例的节点数
        ListNode* dummyHead = new ListNode(); 	// 虚拟头结点
        ListNode* pre = dummyHead;
        for(int i = 0; i < n; ++i){             // 建立单向链表
            cin >> num;
            ListNode* curr = new ListNode();
            pre->m_pNext = curr;
            curr->m_nKey = num;
            pre = curr;
        }
        cin >> k;
        ListNode *slow = dummyHead->m_pNext;    // 快慢指针
        ListNode *fast = dummyHead->m_pNext;
        if(k == 0){								// 倒数第0个数，剪枝
            cout << 0 << endl;
        }else{
            for(int i = 0; i < k; ++i){        	// 快指针比慢指针快k步
                fast = fast->m_pNext;
            }
            while(fast != nullptr){             // 当快指针走到null节点时
                slow = slow->m_pNext;
                fast = fast->m_pNext;
            }
            cout << slow->m_nKey << endl;       // 输出slow节点
        }      
    }
    return 0;
}