链表中倒数第k个结点
链表中倒数第k个结点
输入一个链表,输出该链表中倒数第k个结点。
我用了一个方法,就是新建一个ArrayList存放k个元素,如果没有了就从里面找倒数第k个元素。
/*
public class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}*/
import java.util.ArrayList;
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
if(head==null||k<1){
return null;
}
ArrayList<ListNode>save=new ArrayList<ListNode>(k);
Integer index=0;
Integer count=0;
while(head!=null){
if(save.size()<k){
save.add(head);
}else{
save.set(index,head);
}
index=(index+1)%k;
head=head.next;
count++;
}
if(count<k){
return null;
}
return save.get(index);
}
}
但有一个更好的办法,只用到了指针,不需要存储,空间复杂度是O(1)的:
快指针先往前走k步,注意判断边界,然后快慢一起走,当快指针为none的时候,慢指针走到了倒数第k个节点
public class Solution {
public ListNode FindKthToTail(ListNode head,int k) {
if(k<1||head==null){
return null;
}
ListNode slow=head;
ListNode fast=head;
for(int i=0;i<k-1;i++){
fast=fast.next;
}
if(fast==null){
return null;
}
while(fast.next!=null){
slow=slow.next;
fast=fast.next;
}
return slow;
}
}
本题中的知识点:
- 返回值要的是节点而不是节点的值,就必须要么是原节点,要么new的节点的next也要设置好.不能只返回一个值设置好了的但是next不一样的节点类型
- ArrayList有get和set方法,但是必须先填充(add),否则size就是0,用get或者set会报错
- 本题对于边界情况考察比较重,有:k=0,k大于链表长度,链表为空这几种特殊情况,要考虑到
4.用指针解决链表问题是一种要优先考虑的思路

浙公网安备 33010602011771号