[数据结构与算法]leetcode 剑指 Offer 22. 链表中倒数第k个节点

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

题干

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

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

知识点&算法

链表取倒很容易想到快慢指针的做法，这里因为是取倒数k个，只要将快指针向后k-1位即可。

题解

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        ListNode* slow = head;
        ListNode* fast = head;
        while(--k){
            fast = fast->next;
        }
        while(fast->next != nullptr){
            slow = slow->next;
            fast = fast->next;
        }
        return slow;
    }
};

或者以改变head的代价来节省一个ListNode*的空间

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        ListNode* fast = head;
        while(--k){
            fast = fast->next;
        }
        while(fast->next != nullptr){
            head = head->next;
            fast = fast->next;
        }
        return head;
    }
};