题解 | #链表中倒数最后k个结点#基于vcetor

/**
 * 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) {
        if (pHead == nullptr && k <= 0) return nullptr;
        ListNode* p = pHead;
        vector<ListNode*> arr;
        while (pHead) {
            arr.push_back(pHead);
            pHead = pHead->next;
        }
        int n = arr.size();
        if (n < k) return nullptr;
        else if (n == k) return p;
        else return arr[n - k];
    }
};