链表中倒数第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;
    }
}

本题中的知识点:

  1. 返回值要的是节点而不是节点的值,就必须要么是原节点,要么new的节点的next也要设置好.不能只返回一个值设置好了的但是next不一样的节点类型
  2. ArrayList有get和set方法,但是必须先填充(add),否则size就是0,用get或者set会报错
  3. 本题对于边界情况考察比较重,有:k=0,k大于链表长度,链表为空这几种特殊情况,要考虑到
    4.用指针解决链表问题是一种要优先考虑的思路
posted @ 2020-03-02 09:55  别再闹了  阅读(51)  评论(0)    收藏  举报