22. 链表中倒数第k个节点

原题链接

题目描述

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

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

示例：

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

返回链表 4->5.

---

算法与思路

首先我们需要知道整条链表的长度，记为 \(n\)。

• 当 \(k>n\) 时，链表内倒数第 \(k\) 个数即为 \(nullptr\)，返回 \(nullptr\) 即可。
• 当 \(k \le n\) 时，取 \(k=n-k\)，然后正着遍历链表的第 \(k\) 个数即可。

时间复杂度：最多需要遍历两次链表，所以复杂度为 \(O(n)\)。

---

代码

C++

/**
 * 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) {
        int n = 0;
        ListNode* p = head;
        while (p) {
            n++;
            p = p->next;
        }

        if (k > n) return NULL;

        k = n - k;
        ListNode* q = head;
        while (k-- && q->next) {
            q = q->next;
        }
        return q;
    }
};