输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:
其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
class Solution {
public:
int lengthList(ListNode *head) {
int len = 1;
while (head->next) {
head = head->next;
len++;
}
return len;
}
ListNode *FindKthToTail(ListNode *pHead, int k) {
if (pHead == nullptr || pHead->next == nullptr)
return pHead;
//求链表的长度
int length = lengthList(pHead);
if (length < k)
return nullptr;
ListNode *temp = pHead;
for (int i = 0; i < length - k; i++) {
temp = temp->next;
}
return temp;
}
};