描述
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
进阶:空间复杂度 O(1)O(1),时间复杂度 O(n)O(n)
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:
其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
我的思路
这个题也很简单,也是一个双指针的题目。如果要得到倒数的第k个节点,就先让fast指针走k个节点,slow指针再继续走,等到fast指向空的时候就说明slow指向了倒数第k个元素了。
public ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
ListNode slow = pHead,fast = pHead;
int i = 0;//用于记录fast是否走到了k处
while (fast != null){
if (i < k){
fast = fast.next;
i++;
}else {
fast = fast.next;
slow = slow.next;
}
}
return slow;
}
但上面这个代码没有做边界判断,所以没有通过所有的测试用例。
我原来在力扣写并没有k大于i的case,所以就疏忽了。在牛客运行通的代码如下:
public ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
ListNode slow = pHead,fast = pHead;
int i = 0;//用于记录fast是否走到了k处
while (fast != null){
if (i < k){
fast = fast.next;
i++;
}else {
fast = fast.next;
slow = slow.next;
}
}
if (i < k)
return null;
return slow;
}