[剑指-Offer] 22. 链表中倒数第k个节点(常规解法、特殊情况、鲁棒性)

1. 题目来源

链接：链表中倒数第k个节点
来源：LeetCode——《剑指-Offer》专项

2. 题目说明

输入一个链表，输出该链表中倒数第 k 个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第 1 个节点。例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

示例：

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

3. 题目解析

方法一：先走k - 1步解法

这个想法很简单，也只需要遍历一遍，但是针对链表的题坑点还是很多的：

• head 为空，直接返回 nullptr
• 链表节点数小于 k 时，采用这种解法会出错
• 若 k <= 0 会根据 k 的数据类型，例如 k 为无符号整形时，产生不可预计的错误

书上的坑点给列了这么三种，由于 LeetCode 有些不合法的条件都自行去掉了，但是作为一个优秀的程序员来讲，这些可能会导致代码崩溃的情况一定需要避免。

至于还有逆置链表的写法、有创建 vector<ListNde*> 的写法在此就不列了，效率都不怎么高，也不难，有兴趣可以自行实现。但是更为重要的还是要提升代码的鲁棒性。

参见代码如下：

// 执行用时 :0 ms, 在所有 C++ 提交中击败了100.00%的用户
// 内存消耗 :12.9 MB, 在所有 C++ 提交中击败了100.00%的用户

/**
 * 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 == nullptr) return nullptr;
        int count = 0;
        ListNode* node = head;
        ListNode* tmp = head;
        while (head) {
            head = head->next;
            ++count;
        }
        if (k > count) return nullptr;
        
        while (k) {
            tmp = tmp->next;    
            --k;
        }
        while (tmp) {
            tmp = tmp->next;
            node = node->next;
        }
        return node;
    }
};