【数据结构与算法】链表中倒数最后k个结点

简单题,随便记录一下
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。

数据范围:
0≤n≤10^5,
0≤ai ≤10^9
0≤k≤10^9

要求:空间复杂度 O(n),时间复杂度 O(n)
进阶:空间复杂度 O(1),时间复杂度 O(n)

解法1-----队列

思路很简单,固定存k个节点,遍历到链表最后返回队列里的第一个节点就行,没啥好说的

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) 
    {
        // write code here
        if(pHead == nullptr) return pHead;
        queue<ListNode*> q;
        while(pHead)
        {
            if(q.size() < k) q.push(pHead);
            else
            {
                q.pop();
                q.push(pHead);
            }
            pHead = pHead->next;
        }
        if(q.size() < k) return nullptr;
        return q.front();
    }
};

解法2-----快慢指针

思路也很简单,快指针比慢指针快k步,快指针遍历完了之后返回慢指针的节点就行

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) 
    {
        // write code here
        ListNode* slow = pHead;
        ListNode* fast = pHead;
        for(int i = 0; i < k; ++i)
        {
            if(fast == nullptr) return nullptr;
            fast = fast->next;
        }
        while(fast)
        {
            slow = slow->next;
            fast = fast->next;
        }
        return slow;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值