描述

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

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

数据范围：

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

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

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

img

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

示例1

输入：
{1,2,3,4,5},2
返回值：
{4,5}
说明：
返回倒数第2个节点4，系统会打印后面所有的节点来比较。

示例2

输入：
{2},8
返回值：
{}

题解1

首先求出链表的长度n，从题目的倒数第k个结点，推出是正数的第n - k个结点。若n - k < 0，则返回空值

/**
 * 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
        int len = 0;
        ListNode* p = pHead;
        for (; p != nullptr; p = p->next, len++);
        int step = len - k;
        if (step < 0) {
            return nullptr;
        }
        while (step--) {
            pHead = pHead->next;
        }
        return pHead;
    }
};

题解2

用一个栈处理，由于栈是后进先出的，所以正好符合题意的倒数。倒数第k个结点就意味着栈需要弹出k次即可。

这里有一个细节，栈都弹完了，k还是大于等于0的，则应该返回空值

/**
 * 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
        stack<ListNode*> st;
        ListNode* res;
        if(k <= 0) return nullptr;
        while (pHead != nullptr) {
            st.push(pHead);
            pHead = pHead->next;
        }
        while (k-- && !st.empty()) {
            res = st.top();
            st.pop();
        }
        if (k >= 0)
            return nullptr;
        return res;
    }
};