快慢指针
将慢指针指向头结点,快指针指向其之后k个节点,使得二者之间距离为k。
将它们一起向后移动,直到快指针指向nullptr,此时慢指针就指向倒数第k个结点。
ListNode* getKthFromEnd(ListNode* head, int k) {
if (head == nullptr)
return head;
ListNode *fast = head->next;
ListNode *slow = head;
// 默认k合法
while (--k > 0)
fast = fast->next;
while (fast != nullptr) {
fast = fast->next;
slow = slow->next;
}
return slow;
}
这个方法的问题是题目默认k的输入是合法的,否则需要检查k大于结点数的情况。因此我想到了下面这样递归的方法。
递归
递归地遍历链表,并返回当前结点是链表中倒数第几个结点。如果k不合法则最后返回nullptr即可。
private:
ListNode *result = nullptr;
// 递归的遍历函数
int traverse(ListNode *node, int k) {
// nullptr是倒数第0个结点
if (node == nullptr)
return 0;
// 当前结点是倒数第n个结点
int n = traverse(node->next, k) + 1;
// 当n是k时,找到结点
// 如果一直都没找到,最终result仍然是nullptr
if (k == n)
result = node;
return n;
}
public:
ListNode* getKthFromEnd(ListNode* head, int k) {
if (head == nullptr)
return head;
traverse(head, k);
return result;
}