程序员面试金典第二章：链表（2） 链表中倒数第k个结点

1.1 题目描述

输入一个链表，输出该链表中倒数第k个结点。

1.2解题思路

思路1：用两个指针，一个先走k步，然后两个一起走，每次走一步，直到先走的走到最后，返回后一个指针指向的节点。
时间复杂度O（n） 空间复杂度O（1）

class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        if(pListHead == nullptr)
            return nullptr;
        ListNode * pre = pListHead;
        while(k--) {
            if(pre == nullptr)
                return nullptr;
            pre = pre ->next;
        }
        ListNode * p = pListHead;
        while(pre != nullptr) {
            pre = pre ->next;
            p = p ->next;
        }
        return p;
    }
};

思路2：可以弄一个stack，把所有的数字压栈，弹出的第k个，就是需要的节点（同理，也可以弄一个queue）
时间复杂度O（n）， 空间复杂度O（n）

class Solution {
public:
    ListNode* FindKthToTail(ListNode* pListHead, unsigned int k) {
        if(k <= 0 || pListHead == nullptr)
            return nullptr;
        stack<ListNode *> sta;
        ListNode * p = pListHead;
        while(p != nullptr) {
            sta.push(p);
            p = p ->next;
        }
        if(sta.size() < k)
            return nullptr;
        for(int i = 0; i < k - 1; ++i)
            sta.pop();
        return sta.top();
    }
};

tips:1.判断k和pListHead输入的合法性
2.可能根本没有k个，直接返回nullptr