剑指Offer 22.链表中倒数第k个节点

题目地址

快慢指针

将慢指针指向头结点，快指针指向其之后k个节点，使得二者之间距离为k。
将它们一起向后移动，直到快指针指向nullptr，此时慢指针就指向倒数第k个结点。

ListNode* getKthFromEnd(ListNode* head, int k) {
        if (head == nullptr)
            return head;
        ListNode *fast = head->next;
        ListNode *slow = head;
        // 默认k合法
        while (--k > 0)
            fast = fast->next;
        while (fast != nullptr) {
            fast = fast->next;
            slow = slow->next;
        }
        return slow;
    }

这个方法的问题是题目默认k的输入是合法的，否则需要检查k大于结点数的情况。因此我想到了下面这样递归的方法。

递归

递归地遍历链表，并返回当前结点是链表中倒数第几个结点。如果k不合法则最后返回nullptr即可。

private:
    ListNode *result = nullptr;
    // 递归的遍历函数
    int traverse(ListNode *node, int k) {
    	// nullptr是倒数第0个结点
        if (node == nullptr)
            return 0;
        // 当前结点是倒数第n个结点
        int n = traverse(node->next, k) + 1;
        // 当n是k时，找到结点
        // 如果一直都没找到，最终result仍然是nullptr
        if (k == n)
            result = node;
        return n;
    }
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if (head == nullptr)
            return head;
        traverse(head, k);
        return result;
    }