0
点赞
收藏
分享

微信扫一扫

BM8 链表中倒数最后k个结点

九点韶留学 2022-02-26 阅读 33

描述

输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围: 0 ≤ n ≤ 1 0 5 0 \leq n \leq 10^5 0n105 , 0 ≤ a i ≤ 1 0 9 0 \leq a_i \leq 10^9 0ai109 0 ≤ k ≤ 1 0 9 0 \leq k \leq 10^9 0k109要求:空间复杂度 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
返回值:{}

代码

/**
 * 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||k==0)
            return NULL;
        int n=0,i=0;
        ListNode *p=pHead;
        while(p){
            n++;
            p=p->next;
        }
        if(n<k)
            return NULL;
        while(pHead){
            if(n-i==k)
                break;
            pHead=pHead->next;
            i++;
        }
        return pHead;
    }
};
举报

相关推荐

0 条评论