剑指offer--链表中倒数第k个结点(Java)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/weixin_40244153/article/details/87924890

题目:输入一个链表,输出该链表中倒数第k个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个结点。例如一个链表有6个结点,从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点

思路:第一直觉是先从头开始遍历,计算链表个数n,然后重新遍历,第n-k+1个结点即为所需要的结点。但是需要遍历2次。也可以使用栈来解决,书中的方法则是:设置两个指针,第一个指针先遍历k-1步;从第k步开始,第二个指针指向头结点,两个结点同时往后遍历,当第一个指针到达最后一个结点时,第二个指针指向的正好是倒数第k个结点。(其实感觉还是近似遍历了两次啊。。)

测试算例:1.功能测试(第k个结点分别在链表的中间,头结点和尾结点)
             2.特殊测试(头结点为null,k超出范围)

考点:代码的鲁棒性

实现:

//方法1:利用两个相距为k-1的指针
public ListNode FindKthToTail(ListNode head,int k) {
        if(head==null||k<=0){
            return null;
        }
        ListNode fast=head;
        for(int i=1;i<=k-1;i++){
            if(fast.next!=null){
                fast=fast.next;
            }
            else{
                return null;
            }
        }
        ListNode low=head;
        while(fast.next!=null){
            fast=fast.next;
            low= low.next;
        }
        return low;
    }
//方法2:利用栈
    public ListNode FindKthToTail(ListNode head,int k) {
        if(head==null || k<=0)
            return null;
        int numbOfList=1;
        Stack<ListNode> st = new Stack<ListNode>();
        st.push(head);
        ListNode node=head.next;
        while(node!=null){
            numbOfList++;
            st.push(node);
            node=node.next;
        }
        if(k>numbOfList)
            return null;
        else{
            for(int i=1;i<=k;i++){
                node=st.pop();
            }
            return node;
        }
    }

收获:

1.注意代码的鲁棒性,开始思考前都需要注意特殊输入测试;
2.一个指针遍历链表无法解决问题时,可以考虑使用两个指针来遍历链表:两个指针先后遍历,即让一个指针先走若干步(该题目)、或者两个指针遍历速度不同(如:求链表中的中间结点,可以令一个指针一次走一步,另一个指针一次走两步来实现

猜你喜欢

转载自blog.csdn.net/weixin_40244153/article/details/87924890