返回倒数第 k 个节点

题目描述

来源于 https://leetcode-cn.com/

实现一种算法,找出单向链表中倒数第 k 个节点。返回该节点的值。

注意:本题相对原题稍作改动

示例:

输入: 1->2->3->4->5 和 k = 2
输出: 4

说明:

给定的 k 保证是有效的。

解法一:快慢指针

class Solution {
public:
    int kthToLast(ListNode* head, int k) {
        ListNode *fast = head, *slow = head;
        for(int i=0;i<k;i++){
            fast = fast->next;
        }
        while(fast != nullptr){
            fast = fast->next;
            slow = slow->next;
        }
        return slow->val;
    }
};

解法二:递归

传入一个整型的指针或者引用,初始是值为 k,在递归退出的时候将其减 1,当递归退出到倒数第 k 个节点时,其值自然已经减小为 0 了,此时返回当前节点即可。

class Solution {
public:
    int kthToLast(ListNode* head, int k) {
        ListNode *node = kth_node_to_last(head, k);
        return node->val;
    }

private:
    static ListNode* kth_node_to_last(ListNode* head, int& k){
        if(head == nullptr) return nullptr;
        ListNode* node = kth_node_to_last(head->next, k);
        if(node != nullptr){
            return node;
        }
        
        k--;
        if(k == 0){
            return head;
        }
        
        return nullptr;
    }
};