剑指offer——JZ22 链表中倒数最后k个结点 解题思路与具体代码【C++】

一、题目描述与要求

链表中倒数最后k个结点_牛客题霸_牛客网 (nowcoder.com)

题目描述

输入一个长度为 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,2,3,4,5},2时,对应的链表结构如下图所示:

其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。

示例

示例1:

输入:{1,2,3,4,5},2

返回值:{4,5}

说明:返回倒数第2个节点4,系统会打印后面所有的节点来比较。

示例2:

输入:{2},8

返回值:{}


二、解题思路

思路①【利用链表长度】时间O(n) 空间O(1)

首先我们先获取链表的长度,如果链表的长度len<k,那么就直接返回nullptr,如果链表的长度len==k,那么直接返回头结点即可。如果len>k,我们需要找到链表中的最后k个结点,因此链表可以分成两部分,前面就是num个结点+最后k个结点=len。因此我们可以遍历链表同时更新num表示当前是第几个结点,当num+k==len时,就代表下一个结点就是最后k的结点的"开头",因此直接返回pHead->next。其中最后的while循环也可以改成for循环,也就是循环n-k次。

思路②【利用快慢指针】 时间O(n) 空间O(1)

定义快慢指针,先让快指针fast向后走k步,同时进行判断,如果fast到不了k步,说明链表太短,所以直接返回nullptr。如果可以的话,接着就是让slow和fast同时向后移动,一直到fast指向空结点,返回slow即可


三、具体代码

思路①

cpp 复制代码
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) {
        int len=0; ListNode *p=pHead;
        int num=0;
        while(p){
            len++;
            p=p->next;
        }
        if(len==k) return pHead;
        while(pHead){
            num++;//访问到第几个结点
            if(num+k==len) 
                return pHead->next;
            pHead=pHead->next;
        }
        return nullptr;
    }
};

思路②

cpp 复制代码
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) {
        ListNode *slow=pHead,*fast=pHead;//利用快慢指针
        for(int i=0;i<k;i++){
            if(fast!=nullptr) fast=fast->next;
            else return slow=nullptr;
        }
        //然后slow和fast同时向后走,当fast指向空时,slow所指向的结点就是最后第k个结点
        while(fast){
            slow=slow->next;
            fast=fast->next;
        }
        return slow;
    }
};
相关推荐
爱吃喵的鲤鱼14 分钟前
linux进程的状态之环境变量
linux·运维·服务器·开发语言·c++
懒惰才能让科技进步36 分钟前
从零学习大模型(十二)-----基于梯度的重要性剪枝(Gradient-based Pruning)
人工智能·深度学习·学习·算法·chatgpt·transformer·剪枝
7年老菜鸡41 分钟前
策略模式(C++)三分钟读懂
c++·qt·策略模式
Ni-Guvara1 小时前
函数对象笔记
c++·算法
似霰1 小时前
安卓智能指针sp、wp、RefBase浅析
android·c++·binder
芊寻(嵌入式)1 小时前
C转C++学习笔记--基础知识摘录总结
开发语言·c++·笔记·学习
獨枭1 小时前
C++ 项目中使用 .dll 和 .def 文件的操作指南
c++
霁月风1 小时前
设计模式——观察者模式
c++·观察者模式·设计模式
橘色的喵1 小时前
C++编程:避免因编译优化引发的多线程死锁问题
c++·多线程·memory·死锁·内存屏障·内存栅栏·memory barrier
泉崎1 小时前
11.7比赛总结
数据结构·算法