BM8 链表中倒数最后k个结点
知识点链表
描述
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围:,
,
要求:空间复杂度
,时间复杂度
进阶:空间复杂度
,时间复杂度
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
示例1
输入:
{1,2,3,4,5},2
复制返回值:
{4,5}
复制说明:
返回倒数第2个节点4,系统会打印后面所有的节点来比较。
示例2
输入:
{2},8
返回值:
{}
题解
思路:
- 1. 先找到第k个节点
- 2. 使用一个指针指向头结点,然后将第k个节点和这个指针分别后移,直到后面的节点为空即可
struct ListNode
{
int val;
struct ListNode *next;
ListNode(int x) : val(x), next(nullptr)
{
}
ListNode() = default;
};
ListNode *FindKthToTail(ListNode *pHead, int k)
{
int i = 0;
auto node = pHead;
while (i < k && node != nullptr)
{
node = node->next;
i++;
}
if (i < k)
{
return nullptr;
}
auto pre_node = pHead;
while (node != nullptr)
{
pre_node = pre_node->next;
node = node->next;
}
return pre_node;
}