leetcode 链接：链表中倒数第k个节点

题目

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

示例：

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

解答 & 代码

/**
 * 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 == NULL || k == 0)    // 特殊情况：如果链表为空，or k == 0，直接返回 NULL
            return NULL;

        ListNode* fast = head;        // 快指针
        ListNode* slow = head;        // 慢指针

        // 快指针先走 k 步
        while(fast != NULL && k > 0)
        {
            fast = fast->next;
            --k;
        }
        if(fast == NULL && k > 0)    // 特殊情况：如果 k 的长度大于链表长度，直接返回 NULL
            return NULL;

        // 快慢指针同时走，快指针到末尾（NULL），慢指针指向的位置就是倒数第k个节点
        while(fast != NULL)
        {
            fast = fast->next;
            slow = slow->next;
        }

        return slow;
    }
};

执行结果：

执行结果：通过

执行用时：4 ms, 在所有 C++ 提交中击败了 76.53% 的用户
内存消耗：10.4 MB, 在所有 C++ 提交中击败了 5.59% 的用户