【算法练习】链表中倒数最后k个结点

题源：

牛客

描述

输入一个长度为 n 的链表，设链表中的元素的值为 ai ，返回该链表中倒数第k个节点。

如果该链表长度小于k，请返回一个长度为 0 的链表。

数据范围：0≤n≤105，0≤ai≤109，0≤k≤109

要求：空间复杂度 O(n)，时间复杂度O(n)

进阶：空间复杂度O(1)，时间复杂度O(n)

例如输入{1,2,3,4,5},2时，对应的链表结构如下图所示：

其中蓝色部分为该链表的最后2个结点，所以返回倒数第2个结点（也即结点值为4的结点）即可，系统会打印后面所有的节点来比较。

示例1

输入：

{1,2,3,4,5},2

返回值：

{4,5}

说明：

返回倒数第2个节点4，系统会打印后面所有的节点来比较。

示例2

输入：

{2},8

返回值：

{}

思路：

快慢指针，快指针先走k步，慢指针再走，随后慢指针便指向第k个节点

题解：

/**
 * struct ListNode {
 *  int val;
 *  struct ListNode *next;
 *  ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param pHead ListNode类
     * @param k int整型
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) {
        // write code here
        //统计链表长度 若链表长度 len < k 则，返回0；否则返回链表最后k个值
        //双指针法
        ListNode* fast = pHead;
        ListNode* slow = pHead;
        //1. 若链表为空 ，返回空
        if (pHead == nullptr)
            return nullptr;//为空
        //2. fast指针先走k步
        while (k--) {
            if (fast == nullptr) return nullptr;
            fast = fast->next;
        }
        //3. 快慢指针同时走
        while (fast != nullptr) {
            fast = fast->next;
            slow = slow->next;
        }
        return slow;
    }
};