问题:输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
要求:空间复杂度 O(n),时间复杂度O(n)
进阶:空间复杂度 O(1),时间复杂度 O(n)
思路:
1、暴力,直接两个循环解决
代码:
public static ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
ListNode a=pHead;
ListNode b=pHead;
int a1=0;int b1=0;
while (a!=null){
a1++;
a=a.next;
}
if(k>a1){
return null;
}
while (b!=null){
if (a1==k){
return b;
}
a1--;
b=b.next;
}
return null;
}
2、设置两个指针,第一个指针走了k步以后,第二个指针在开始向下遍历,
public static ListNode FindKthToTail (ListNode pHead, int k) {
// write code here
//优化
ListNode a=pHead;
ListNode b=pHead;
int i=0;
while (b!=null){
if(i>=k){
a=a.next;
}
i++;
b=b.next;
}
if(i<k){
return null;
}
return a;
}