[题解]剑指 Offer 22. 链表中倒数第k个节点(C++)

题目

输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。

例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

示例：

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/lian-biao-zhong-dao-shu-di-kge-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

TSSN的想法是先遍历一遍链表，得到其长度n，然后输出顺数第n - k个节点。
直接上神器双指针，fast和slow开始都指向head，fast先走k步，然后slow和fast同时向后遍历，当fast遍历完链表指向空节点的时候，slow即为倒数第k个节点。另外，加上一些边界条件判定，若链表为空直接返回空指针；若k比链表长度还长，那么倒数第k个节点就不存在了，也返回空指针。
时间复杂度O(n)，空间复杂度O(1)。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if(!head) return head;
        ListNode *slow = head, *fast = head;
        while(fast && k > 0)
        {
            fast = fast->next;
            --k;
        }
        if(k > 0) return nullptr;
        while(fast != nullptr)
        {
            slow = slow->next;
            fast = fast->next;
        }
        return slow;
    }
};