【剑指Offer刷题】——链表中倒数第k个节点（22）

Leetcode刷题地址：https://leetcode-cn.com/problemset/lcof/

题目

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

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

示例：

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

返回链表 4->5.

Solution 1

第一次遍历计算链表长度，第二次遍历到n-k的位置。

时间：O(n)，假设输入链表长度为n；空间：O(1)

ListNode* getKthFromEnd(ListNode* head, int k) {
    ListNode* temp = head;
    int n = 0;
    while (temp != NULL) {
        n++;
        temp = temp->next;
    }
    temp = head;
    for (int i=1; i<=n-k; i++) {
        temp = temp->next;
    }
    return temp;
}

结果：

Solution 2

用一个数组存储链表中指向每一个节点的指针，返回数组下标为n-k的指针。

时间：O(n)；空间：O(n)

ListNode* getKthFromEnd(ListNode* head, int k) {
    ListNode* temp = head;
    vector<ListNode*> v;

    while (temp != NULL) {
        v.push_back(temp);
        temp = temp->next;
    }
    return v[v.size()-k];
}

Solution 3

快慢指针解法。用一个快指针先前进k个节点，然后让快指针和慢指针同步前进，等到快指针到达链表尾部，此时慢指针指向了倒数第k个节点。

时间：O(n)，假设输入链表长度为n；空间：O(1)

ListNode* getKthFromEnd(ListNode* head, int k) {
    if (!head || k <= 0) return NULL;
    ListNode *quick = head, *slow = head;
    while (k--) {
        if (!quick) return NULL;        // 当链表长度不足k时
        quick = quick->next;
    }
    while (quick) {
        quick = quick->next;        // 快指针
        slow = slow->next;          // 慢指针
    }
    return slow;
}

结果：（按理说应该要比solution1好一些…